我想在 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
的填充外观如何?
谢谢你。