如何将我的数据集中的每个第5个样本分配为测试数据集,将其余样本分配为训练数据集？

Question

我有一个包含133个样本的数据集(133行和2个coloumns(属性)).我需要从1(第1,第6,第11,第16 ......等)开始分配每个第5个样本作为测试数据集,其余作为训练数据集.我该怎么做？

Answer 1

使用Maltab中的冒号运算符:

testset = dataset(1:5:end,:)
trainingset = setdiff(dataset, testset, 'rows')

你也可以这样做setdiff:

testset = dataset(1:5:end,:)
trainingset = dataset;
trainingset(1:5:end, :) = [];