java - Spark 管道上的 Deeplearning4j : What are the arguments to the org. apache.spark.ml.PredictionModel.predict 方法

Question

我正在尝试添加用 deeplearning4j 编写的情感分析程序到 Spark 管道。然后我必须重写方法 org.apache.spark.ml.PredictionModel. predict()使用 RNN 进行预测我创建的模型。据我了解，此方法的参数是特征行，并且该行的标签是预测的。但是，在情感分析程序中，特征应该是一个二维数组，因为每一行都包含一个 vector 列表，如 [[0.0011181544391649161,0.0025584171060651644,0.01754946247376411,-0.006530340570481004,0.003487414946750136,0.004426218948032432,0.00404,0.002611281607120172,0.006444432718879956,-0.012260229877306768,0.002399729592556043]]

但是我得到一个一维数组 [0.003356837383,0.0074654373,...]作为预测方法的输入。您能解释一下为什么会发生这种情况吗？您能解释一下预测方法的参数是什么吗？

Answer 1

我们的 usingrnns 页面可能会给您一些想法: http://deeplearning4j.org/usingrnns

dl4j 的内部表示无论如何都需要与此匹配。