怎么划分训练集和测试集
时间: 2023-08-23 17:04:17 浏览: 99
python 划分数据集为训练集和测试集的方法
以下是MATLAB划分训练集、验证集和测试集的代码示例:
假设我们有一个数据集X和对应的标签y,我们想将其划分为训练集、验证集和测试集,比例分别为70%、15%和15%。
首先,我们需要将数据集和标签随机打乱,以保证划分的随机性:
```
idx = randperm(size(X,1));
X = X(idx,:);
y = y(idx,:);
```
然后,我们可以按照比例划分数据集:
```
% 划分训练集
train_ratio = .7;
train_size = floor(train_ratio * size(X,1));
X_train = X(1:train_size,:);
y_train = y(1:train_size,:);
% 划分验证集
val_ratio = .15;
val_size = floor(val_ratio * size(X,1));
X_val = X(train_size+1:train_size+val_size,:);
y_val = y(train_size+1:train_size+val_size,:);
% 划分测试集
test_ratio = .15;
test_size = floor(test_ratio * size(X,1));
X_test = X(train_size+val_size+1:end,:);
y_test = y(train_size+val_size+1:end,:);
```
最后,我们可以将划分好的数据集用于训练、验证和测试模型。
阅读全文