对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
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)