matlab划分训练集验证和测试集代码_Matlab随机划分训练集和测试集

%导入数据集

clear;

clc;

load wine_SVM;           %数据集为wine，数据集标签为wine_labels

Data = wine;                 %对数据集实例化

num_test = 78;              %设置测试集样本数为78，训练集样本为100

[ndata, D] = size(Data);     %ndata表示数据集样本数，D是特征数

R = randperm(ndata);         %1到ndata这些数随机打乱得到的一个随机数字序列作为索引

Xtest = Data(R(1:num_test),:);    %以索引的前78个数据点作为测试样本Xtest

test_wine_labels = wine_labels(R(1:num_test),:);      % 设置测试集样本标签

test_wine_labels = double( test_wine_labels );

R(1:num_test) = [];

Xtrain = Data(R,:);             %剩下的100个样本数据作为训练样本Xtrain

num_train = size(Xtrain,1);     %num_train：训练样本数

train_wine_labels = wine_labels(R,:);          % 设置训练集样本标签

train_wine_labels = double( train_wine_labels );