2016-07-12 56 views
0

我在MLLib階決策樹:保存MLLib決策樹模型

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

我想這個模型保存到磁盤或HDFS。當我鍵入tree.save(...)時,會引發一個編譯錯誤,基本上指向模型的這種方法。

任何指針都會非常有幫助。謝謝!

回答

0

DecisionTreeClassificationModel中沒有保存方法,因此拋出以下錯誤。爲了將它保存到磁盤或hdfs,你需要有一個你想要保存到磁盤的對象的RDD。爲此,您可以創建一個RDD [DecisionTreeClassificationModel],然後使用相應的可用保存方法。

+0

我不能看你怎麼做。你能給我一個scala代碼嗎?謝謝。 – user3553070

0

save方法屬於:

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

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