保存MLLib决策树模型

Question

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

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

我想将此模型保存到磁盘或hdfs。当我输入tree.save（...）时，它会抛出一个编译错误，基本上指向模型没有这样的方法。

任何指针都会非常有用。谢谢！

Answer 1

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

Answer 2

save方法属于：

  

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

DecisionTreeClassificationModel

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