Deeplearning4j - 如何填充 RNN INDArray?
Deeplearning4j - How to fill RNN INDArray?
我想在 Deeplearning4j 中编写一个 RNN 用于股市预测,但我正在努力创建和填充 3 维 INDArrays
。
所以如果我有以下时间序列:
1 2 3 4 5 6 7 8 9 10
我想使用 5 个值作为输入并预测第 6 个值:
Input: 1 2 3 4 5 TeachingInput: 6
Input: 2 3 4 5 6 TeachingInput: 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');
这是正确的吗?如果是这样,features、labels、featuresMask 的填充 INDArrays
看起来如何] 和 labelsMask?
谢谢。
您可能需要查看 deeplearning4j。org/usingrnns
如果您出于某种原因需要自己动手,请查看我们在 sequencerecordreaderdatasetiterator 中所做的工作。
如果可以的话,我强烈建议您使用 datavec。我们已经为您解决了这个问题。
内部。
我想在 Deeplearning4j 中编写一个 RNN 用于股市预测,但我正在努力创建和填充 3 维 INDArrays
。
所以如果我有以下时间序列:
1 2 3 4 5 6 7 8 9 10
我想使用 5 个值作为输入并预测第 6 个值:
Input: 1 2 3 4 5 TeachingInput: 6
Input: 2 3 4 5 6 TeachingInput: 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');
这是正确的吗?如果是这样,features、labels、featuresMask 的填充 INDArrays
看起来如何] 和 labelsMask?
谢谢。
您可能需要查看 deeplearning4j。org/usingrnns
如果您出于某种原因需要自己动手,请查看我们在 sequencerecordreaderdatasetiterator 中所做的工作。
如果可以的话,我强烈建议您使用 datavec。我们已经为您解决了这个问题。
内部。