在deeplearning4j中是否有检查我输入数据的维度?我正在尝试对股票数据集进行时间序列预测,当我尝试编译模型时,总是收到相同的错误消息:
Exception in thread "main" java.lang.IllegalStateException: 3D input expected to RNN layer expected, got 2
我的模型的代码如下所示,但我认为问题出在我的数据上,而不是模型本身:
MultiLayerConfiguration config = new NeuralNetConfiguration.Builder()
.optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
.weightInit(WeightInit.XAVIER)
.list()
.layer(0, new LSTM.Builder().activation(Activation.TANH).nIn(105).nOut(600).build())
.layer(1, new RnnOutputLayer.Builder(LossFunctions.LossFunction.MSE)
.activation(Activation.TANH).nIn(600).nOut(1).build())
.build();
是否有方法确保我输入到模型中的dataset/DataSeterator对象使用了正确的维度?谢谢你的帮助。
暂无答案!
目前还没有任何答案,快来回答吧!