matlab cvpartition函数,Matlab: cvpartition()函数实现交叉验证

下面的代码是将一组数据进行10折交叉划分，并将划分好的训练集合测试集存放在一个文件夹下面：

% Note: this code should be run from orca/src/code-examples

clear param;

if (exist ('OCTAVE_VERSION', 'builtin') > 0)

pkg load statistics

end

% Load data

ERAData = csvread('../../exampledata/ERA.csv');

% Check the first 20 rows

ERAData(1:20,:);

% Extract targets

targets = ERAData(:,end);

% Generate k fold partitions

k=10;

CVO = cvpartition(targets,'KFold',k);

% Prepare filesystem

nameDataset = 'era';

rootDir = fullfile('..', '..', 'exampledata', '10-fold', nameDataset);

mkdir(rootDir);

rootDir = fullfile(rootDir,'matlab');

mkdir(rootDir);

% For each fold

for ff = 1:k

if (exist ('OCTAVE_VERSION', 'builtin') > 0)

trIdx = training(CVO,ff);

teIdx = test(CVO,ff);

else

trIdx = CVO.training(ff);

teIdx = CVO.test(ff);

end

dlmwrite(fullfile(rootDir,sprintf('train-%s.%d',nameDataset,ff-1)),ERAData(trIdx,:),' ');

dlmwrite(fullfile(rootDir,sprintf('test-%s.%d',nameDataset,ff-1)),ERAData(teIdx,:),' ');

end

关于 cvpartition()函数的使用说明请参考：

https://ww2.mathworks.cn/help/stats/cvpartition.html