matlab划分数据集代码
时间: 2023-06-03 11:02:52 浏览: 114
我可以通过以下代码将数据集划分为训练集、验证集和测试集:
```
% 导入数据集
data = csvread('data.csv');
% 划分数据集
train_ratio = 0.7; % 训练集比例
val_ratio = 0.15; % 验证集比例
test_ratio = 0.15; % 测试集比例
n = size(data, 1); % 数据集大小
rand_indices = randperm(n); % 随机排序索引
train_indices = rand_indices(1:round(train_ratio*n)); % 训练集索引
val_indices = rand_indices(round(train_ratio*n)+1:round((train_ratio+val_ratio)*n)); % 验证集索引
test_indices = rand_indices(round((train_ratio+val_ratio)*n)+1:end); % 测试集索引
% 保存数据集
train_data = data(train_indices, :);
val_data = data(val_indices, :);
test_data = data(test_indices, :);
csvwrite('train_data.csv', train_data);
csvwrite('val_data.csv', val_data);
csvwrite('test_data.csv', test_data);
```
这段代码将数据集从data.csv中导入,然后随机打乱索引,根据三个比例将数据集分成训练集、验证集和测试集,并保存到三个csv文件中。
阅读全文