我的数据在我的spark
dataframe
(df)中有24个特征,第25列是我的目标变量。我想在此dataset
上安装我的型号,其输入格式为org.nd4j.linalg.api.ndarray.INDArray, org.nd4j.linalg.dataset.Dataset
或org.nd4j.linalg.dataset.api.iterator.DataSetIterator
。如何将我的dataframe
转换为所需的类型?输入火花数据帧到DeepLearning4J模型
我也尝试使用管道方法直接向模型输入火花数据帧。但dl4j-spark-ml的sbt依赖不起作用。我的build.sbt文件是:
scalaVersion := "2.11.8"
libraryDependencies += "org.deeplearning4j" %% "dl4j-spark-ml" % "0.8.0_spark_2-SNAPSHOT"
libraryDependencies += "org.deeplearning4j" % "deeplearning4j-core" % "0.8.0"
libraryDependencies += "org.nd4j" % "nd4j" % "0.8.0"
libraryDependencies += "org.nd4j" % "nd4j-native-platform" % "0.8.0"
libraryDependencies += "org.nd4j" % "nd4j-backends" % "0.8.0"
libraryDependencies += "org.apache.spark" %% "spark-core" % "2.0.1"
libraryDependencies += "org.apache.spark" %% "spark-sql" % "2.0.1"
有人可以从这里引导我吗?提前致谢。
我试过使用'“org.deeplearning4j”%%“dl4j-spark-ml”%“0.8.0_spark_2-SNAPSHOT” 。但它仍然没有工作。我编辑完整的build.sbt文件的问题。请检查一次。 – Ishan
我说0.8。* 1 *让我更明确: https://oss.sonatype.org/内容/重positories/snapshots/org/deeplearning4j/dl4j-spark-ml_2.11/0.8.1_spark_2-SNAPSHOT /它绝对存在。你不应该遇到任何问题。 ^^^ –
我不得不在“https://oss.sonatype.org/content/repositories/snapshots/”使用'resolvers + =“scala-tools.org”来使它工作。谢谢你告诉我这件事。但我想要的是使用 - org.deeplearning4j.spark.ml.classification.NeuralNetworkClassification,但似乎这个类不可用。你有什么想法吗?我想要的只是将我的火花数据框输入到dl4j模型中。 – Ishan