首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >保存MLLib决策树模型

保存MLLib决策树模型
EN

Stack Overflow用户
提问于 2016-07-13 01:16:46
回答 2查看 508关注 0票数 0

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

代码语言:javascript
复制
    val tree = model.stages(1).asInstanceOf[DecisionTreeClassificationModel]

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

任何指针都会很有帮助。谢谢!

EN

回答 2

Stack Overflow用户

发布于 2016-07-13 01:55:04

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

票数 0
EN

Stack Overflow用户

发布于 2016-07-13 17:38:15

save方法属于:

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

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

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/38335250

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档