加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
train_val_split.m 796 Bytes
一键复制 编辑 原始数据 按行查看 历史
李星毅 提交于 2020-07-01 23:38 . First commit
function [train_data, val_data] = train_val_split(...
train_data, train_num)
for i = 1:size(train_data, 2)
% 随机选择一个起点,需要保证这train_num个训练集是连续的
train_start = randi(size(train_data(i).features, 1)-train_num+1);
train_range = train_start:(train_start + train_num-1);
val_range = setdiff(1:size(train_data(i).features, 1), train_range);
x_train = train_data(i).features(train_range, :);
y_train = train_data(i).resTime(train_range);
x_val = train_data(i).features(val_range, :);
y_val = train_data(i).resTime(val_range);
train_data(i).features = x_train;
train_data(i).resTime = y_train;
val_data(i).features = x_val;
val_data(i).resTime = y_val;
end
end
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化