使用DL4J，ND4J和火花來訓練和評估神經網絡

Question

我正嘗試使用deeplearning4j，nd4j和spark創建，訓練和評估一個神經網絡。我能夠在RDD的形式養活神經網絡的數據，但是，當我嘗試評估神經網絡如何與這行代碼的培訓：

lazy val evaluation_Regression_RMSE = 
trainLayer.evaluateRegression(testingDfConverted).rootMeanSquaredError(0) 
println(evaluation_Regression_RMSE) 

我收到此錯誤：

org.nd4j.linalg.exception.ND4JIllegalStateException：op.X長度應該是 等於op.Y長度：[[2,1,2,1,1,0,1,99]]！= [[2 ，1，1，1，1，0，1，99]在 org.nd4j.linalg.cpu.nativecpu.ops.NativeOpExecutioner.exec（NativeOpExecutioner.java:475）

有誰知道發生了什麼這裏以及如何o修復這個錯誤？

Answer 1

您的標籤和輸出數組大小不匹配。它顯示那裏你有一個1 x 2對一個1 x 1.