MATLAB的cvpartition函数用法


1. 函数作用

cvpartition 将数据集划分为训练集和测试集,支持多种交叉验证方法,包括:

  • Hold-Out验证:单次划分(如70%训练,30%测试)
  • K折交叉验证:数据分为K个子集,依次用其中一个作为测试集
  • 留一法(Leave-One-Out):每次留一个样本作为测试集
  • 分层交叉验证:保持不同类别的样本比例一致

2. 基本语法

matlab 复制代码
c = cvpartition(n, 'Method', value, ...)
c = cvpartition(group, 'Method', value, ...)  % 分层选项

3. 参数说明

参数 说明
n 样本数量
group 类别标签向量(用于分层交叉验证)
'Method' 验证方法:'HoldOut', 'KFold', 'LeaveOut'
'HoldOut'参数 测试集比例 p(默认0.1)
'KFold'参数 折数 k(默认10)

4. 常用方法示例

Hold-Out 验证
matlab 复制代码
data = rand(100, 5); % 100个样本,5个特征
c = cvpartition(100, 'HoldOut', 0.3); % 30%作为测试集

trainIdx = training(c); % 逻辑索引(训练集)
testIdx = test(c);      % 逻辑索引(测试集)

trainData = data(trainIdx, :);
testData = data(testIdx, :);
K折交叉验证(5折)
matlab 复制代码
c = cvpartition(100, 'KFold', 5);
for i = 1:c.NumTestSets
    trainIdx = training(c, i); % 第i折的训练索引
    testIdx = test(c, i);      % 第i折的测试索引
    % 在此训练和验证模型
end
分层K折交叉验证
matlab 复制代码
labels = [ones(50,1); zeros(50,1)]; % 类别标签(二分类)
c = cvpartition(labels, 'KFold', 5); % 保持各类比例

% 查看每折的类别分布
disp(c.TestSize); % 每折测试集样本数
留一法(LOO)
matlab 复制代码
c = cvpartition(100, 'LeaveOut');
for i = 1:c.NumTestSets
    trainIdx = training(c, i);
    testIdx = test(c, i); % 每次仅1个测试样本
end

5. 对象属性与方法

  • 属性 :
    • c.NumTestSets:测试集数量(K折时为K)
    • c.TrainSize:各训练集大小
    • c.TestSize:各测试集大小
  • 方法 :
    • training(c, i):获取第i折的训练索引
    • test(c, i):获取第i折的测试索引
    • repartition(c):重新随机化分区(仅Hold-Out和KFold)

6. 注意事项

  • 随机种子 :使用 rng(seed) 确保结果可重复。
  • 分层必要性:分类问题建议使用分层选项,避免类别分布不均。
  • 大数据慎用LOO:留一法计算成本高,样本量大时改用K折。

相关推荐
Evand J21 分钟前
【电机滤波例程2】负载转矩增广扩展卡尔曼滤波(EKF)原理与MATLAB例程:五维PMSM状态与负载阶跃估计。订阅专栏后可查看完整代码
开发语言·matlab·电机·ekf·卡尔曼滤波
weixin_307779138 小时前
C++代码实现MATLAB中的dlarray函数功能
开发语言·c++·算法·matlab
Evand J1 天前
【MATLAB例程】三维RRT+APF路径规划与AOA-TDOA融合定位算法。附代码的下载链接
算法·matlab·路径规划·rrt·tdoa·apf·aoa
Evand J1 天前
【MATLAB例程】三维RRT+APF避障路径规划与到达角(AOA)定位算法|三维路径优化与定位仿真例程
算法·matlab·路径规划·代码·定位·rrt·aoa
菜鸟~noob2332 天前
【太空电子战】午夜锤行动中的电磁静默区:可逆干扰的实战验证【含matlab代码】
linux·开发语言·matlab·电子战
Evand J2 天前
【MATLAB例程】三维快速扩展随机树(RRT)路径规划与到达角(AOA)定位算法|三维避障+定位仿真。附下载链接
开发语言·算法·matlab·路径规划·定位
Zootopia6263 天前
多架 eVTOL集群排班调度方案思考
人工智能·算法·数学建模·matlab·动态规划·无人机·evtol
Evand J3 天前
【MATLAB例程】三维快速扩展随机树(RRT)路径规划与到达角(AOA)、到达时间差(TDOA)融合定位,附下载链接
开发语言·matlab·代码·定位·例程
Evand J3 天前
【MATLAB例程】三维A*路径规划与TOA-AOA-TDOA融合定位算法。附下载链接
数据库·算法·matlab
水桶学习吧_3 天前
电网负载无功MATLAB仿真
matlab·cnn·simulink