2

我想在 Deeplearning4j 中编写一个用于股票市场预测的 RNN,但我正在努力创建和填充 3 维INDArrays。因此,如果我有以下时间序列:

1 2 3 4 5 6 7 8 9 10

我想使用 5 个值作为输入并预测第 6 个值:

输入:1 2 3 4 5教学输入:6

输入:2 3 4 5 6教学输入:7

...

我会这样填写INDArrays

int numExamples = 5; //1-5, 2-6,..., 5-9
int inputSize = 5; 
int timeSeriesLength = 10;
INDArray features =  Nd4j.create(new int[]{numExamples,inputSize,timeSeriesLength}, 'f');

int outputSize = 1;
INDArray labels =  Nd4j.create(new int[]{numExamples,outputSize,timeSeriesLength}, 'f');

它是否正确?如果是这样,特征标签特征掩码和标签掩码INDArrays的填充外观如何?

谢谢你。

4

1 回答 1

0

您可能需要查看 deeplearning4j.org/usingrnns

如果您出于某种原因需要自己动手,请查看我们在 sequencerecordreaderdatasetiterator 中所做的事情。

如果可以的话,我强烈建议您使用 datavec。我们已经为您解决了这个问题。

查看 https://github.com/deeplearning4j/deeplearning4j/blob/master/deeplearning4j-core/src/main/java/org/deeplearning4j/datasets/datavec/SequenceRecordReaderDataSetIterator.java

内部结构。

于 2016-09-01T04:15:52.817 回答