python-拆分顺序数据以进行训练和测试

时间:2020-04-22 11:06:36

标签: python

具有以下数据形状(25、4283、17),包含25个批次的4283个样本,每个样本具有17个特征。 我将批次分为20个训练批次和5个测试批次,如下所示:

train = sequences[:20]
test = sequences[20:]

每行数据中的最后一个索引是批处理目标,对所有样本均重复:

[
   [
     [1,2,3....TARGET_VALUE_1],
     [1,2,3....TARGET_VALUE_1] <- the target of the batch is the last value
     ...
   ],
   [
     [4,5,6....TARGET_VALUE_2],
     [4,5,6....TARGET_VALUE_2] <- the target of the batch is the last value
     ...
   ],
]

我需要的是火车的20个目标值和测试的5个值

[TARGET_VALUE_1, TARGET_VALUE_2 ... TARGET_VALUE_20]
and
[TARGET_VALUE_21,...TARGET_VALUE_25]

任何帮助将不胜感激

0 个答案:

没有答案