deeplearning4j:如何为使用lstm的rnn准备这些数据?

hmmo2u0o  于 2021-06-30  发布在  Java
关注(0)|答案(1)|浏览(401)

在我的代码中,我将数据从一个源下载到一个csv文件,然后对其应用一个转换过程,然后将其写入最终的csv文件。此时,我的一行数据如下所示:
45.414001,10358500,45.698002,44.728001,0.0
第一列是我要预测的数据,最后一列(带0的那列)现在只是一个占位符,它将是一个双数。使用deeplearning4j,然后将这些数据从csv文件加载到一个recordreader中。下面是它的样子:

RecordReader recordReader = new CSVRecordReader(numSkipLines);

recordReader.initialize(new FileSplit(inputPath));

所以我的问题是,我下一步该怎么办?我想把这个数据和rnn lstm模型一起使用,这个模型将预测未来的第一列。我下一步该怎么办?

eqoofvh9

eqoofvh91#

通常,根据您有多少数据/行,您可以将训练数据和测试数据分开。测试数据通常会比训练数据小一些,因为测试只是为了看看模型是否能有效地预测。
训练数据应该在较小的训练集和验证集之间进行分割。您应该能够使用验证集来查看经过多少个时期/轮次的训练后,您是否感到过适/不适。你想训练到模型刚开始过度拟合的时候。

相关问题