Deeplearning4j -如何填充RNN INDArray？

Question

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

%1%2%3%4%5%6%7%8%9 10

我想用5个值作为输入，并预测第6个值：

TeachingInput：输入：1 2 3 4 56 TeachingInput：输入：2 3 4 5 67

..。

我会像这样填写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对于特性()、标签( labels )、和labelsMask的外观如何？

谢谢。

Answer 1

您可能想要查看decplearning4j.org/usingrnns。

如果您由于某种原因需要自己动手，请看我们在序列编辑器中所做的事情。

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

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

内饰。