Deeplearning4j - 如何填写RNN INDArray？

Question

我想在Deeplearning4j中为股票市场预测编写一个RNN，但我正在努力创建和填充三维INDArrays。 如果我有以下时间序列：

  

1 2 3 4 5 6 7 8 9 10

我希望使用5个值作为输入并预测第6个值：

  

输入：1 2 3 4 5 TeachingInput ：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');

这是对的吗？如果是这样，那么功能，标签， featuresMask 和 labelsMask 的填充INDArrays如何？ >

谢谢。

Answer 1

您可以查看deeplearning4j.org/usingrnns

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

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

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

内部构件。