python-拆解sklearn中决策树

python-拆解sklearn中决策树,第1张

获取树结构实体

scikit-learn中DecisionTreeClassifier/Regressor的实例调用.tree_属性可以得到树结构。
参考sklearn的决策树的官方说明sklearn.tree.DecisionTreeClassifier(不过里面说的help(sklearn.tree._tree.Tree)似乎不管用)

获取决策树基本信息

node总数可以用model.tree_.node_count得到。

children_left = model.tree_.children_left
children_right = model.tree_.children_right
feature = model.tree_.feature
threshold 

欢迎分享,转载请注明来源:内存溢出

原文地址: http://www.outofmemory.cn/langs/743213.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-29
下一篇 2022-04-29

发表评论

登录后才能评论

评论列表(0条)

保存