matlab将数据集分成训练集和测试集,Matlab实现 把数据集X分割成训练集和测试集...

【注意：本文写得不好，如有需求请参考其他好的文章！！！】

把数据集分割成训练集和测试集，要求训练集和测试集中的类别比例与原数据集相同，训练集的比例一般大于等于0.7。函数输入参数为数据集矩阵X、对应的类标签y、类别个数k和训练集的比例ratio，返回训练集X_train和对应的类标签y_train 测试集X_test和对应的类标签y_test。

Matlab代码如下：

function [X_train, y_train, X_test, y_test] = split_train_test(X, y, k, ratio)

%SPLIT_TRAIN_TEST 分割训练集和测试集

% 参数X是数据矩阵 y是对应类标签 k是类别个数 ratio是训练集的比例

% 返回训练集X_train和对应的类标签y_train 测试集X_test和对应的类标签y_test

m = size(X, 1);

y_labels = unique(y); % 去重，k应该等于length(y_labels)

d = [1:m]';

X_train = [];

y_train= [];

for i = 1:k

comm_i = find(y == y_labels(i));

if isempty(comm_i) % 如果该类别在数据集中不存在

continue;

end

size_comm_i = length(comm_i);

rp = randperm(size_comm_i); % random permutation

rp_ratio = rp(1:floor(size_comm_i * ratio));

ind = comm_i(rp_ratio);

X_train = [X_train; X(ind, :)];

y_train = [y_train; y(ind, :)];

d = setdiff(d, ind);

end

X_test = X(d, :);

y_test = y(d, :);

end