python - “DecisionTreeClassifier”对象没有属性 'export_graphviz'

Question

我正在使用 python sklearn RandomForestClassifier 并尝试导出决策树。

基本代码如下:

from sklearn import tree

with open(dot_file_name, 'w') as my_file:
    tree.export_graphviz(tree1, out_file = my_file,feature_names = feature_names)

运行python脚本后，出现以下错误:属性错误:“DecisionTreeClassifier”对象没有属性“export_graphviz”

我使用的是Python 2.7。是因为python版本的原因吗？我必须使用 python 3.0 吗？

Answer 1

那是因为您在某处使用了树名称作为DecisionTreeClassifier的名称。在那里使用另一个名称。