Scala Spark負載保存的MLlib模型

Question

我已經在我的S3路徑中保存了一個隨機森林模型，現在我想加載它。 但是，我得到一個錯誤，即該方法不存在。

代碼（保存模型有效）：

import org.apache.spark.ml.classification.RandomForestClassifier

    val rfClassifier = new RandomForestClassifier()
      .setImpurity("gini")
      .setMaxDepth(8)
      .setNumTrees(200)
      .setFeatureSubsetStrategy("auto")
      .setSeed(18)

   val rfModel = rfClassifier.fit(trainingFeatures)
    rfModel
    .write
    .overwrite()
    .save(<MY S3 PATH>)

代碼（加載模型不起作用）：

val rfmodel = RandomForestClassifier.load(<MY S3 PATH>)
)

錯誤：

java.lang.NoSuchMethodException: org.apache.spark.ml.classification.RandomForestClassificationModel.<init>(java.lang.String)

不知道為什么存在加載方法時會發生此錯誤

Answer 1

您應該加載RandomForestClassificationModel而不是RandomForestClassifier 。

用。。。來代替：

val rfmodel = RandomForestClassificationModel.load(<MY S3 PATH>)

更多關於模型持久這里