0

此代码返回特定列的决策树,但没有多少切片或只是将其留空允许我获得所有列的组合决策树。

from sklearn.tree import export_graphviz
import pydot


tree = classifier.estimators_[3]


export_graphviz(tree, out_file = 'tree.dot', feature_names = feature_list, rounded = True, precision = 1)

# Use dot file to create a graph
(graph, ) = pydot.graph_from_dot_file('tree.dot')

# Write graph to a png file
graph.write_png('tree.png')

我知道这与此有关

    tree = classifier.estimators_[3]

我怎么做?

4

0 回答 0