保存MLLib决策树模型

Question

我在MLLib scala中有一个决策树：

    val tree = model.stages(1).asInstanceOf[DecisionTreeClassificationModel]

我想将此模型保存到磁盘或hdfs。 当我键入tree.save（...）时，它会引发编译错误，该错误基本上指向该模型没有这种方法。

任何指针将大有帮助。 谢谢！

Answer 1

DecisionTreeClassificationModel中没有保存方法，因此引发以下错误。 为了将其保存到磁盘或hdfs，您需要具有要保存到磁盘的对象的RDD。 为此，您可以创建一个RDD [DecisionTreeClassificationModel]，然后使用相应的可用保存方法。

Answer 2

save方法属于：

org.apache.spark.mllib.tree.model.DecisionTreeModel

不是org.apache.spark.ml.classification DecisionTreeClassificationModel