导出/打印随机森林决策树/&#39；随机森林分类器对象没有属性&#39；树_&#39； [英] Export / Plot Random Forest Decision Tree / 'RandomForestClassifier' object has no attribute 'tree_'

问题描述

大家晚上好

本文的目标是能够从随机决策树过程中绘制决策树。在运行不同的选项后，我总是得到下一个错误：‘RandomForestVinator’对象没有属性‘tree_’

非常感谢任何帮助/代码示例/想法或链接，以便能够解决此问题。

在下一组代码中，我是如何绘制常规/常规决策树的。

clf_SMOTE1 = DecisionTreeClassifier(criterion='entropy',max_depth=4, min_samples_leaf=7)
clf_SMOTE1
    
clf_SMOTE1 = clf_SMOTE1.fit(X_train, Y_train)
    
a = df.columns[6:]
dot_data = tree.export_graphviz(clf_SMOTE1, out_file=None, filled=False, feature_names= a)
graphviz.Source(dot_data)

在接下来的几行中，我尝试了不同的尝试，但都没有结果。

clf_SMOTE2 = RandomForestClassifier(criterion='entropy', bootstrap = True, max_depth=4, min_samples_leaf=7)
clf_SMOTE2
    
clf_SMOTE2 = clf_SMOTE2.fit(X_train, Y_train)

a = df.columns[6:]
dot_data_2 = tree.export_graphviz (clf_SMOTE2, out_file=None, feature_names = a, precision = 2, filled = False)
graphviz.Source(dot_data_2)

选项2：

clf_SMOTE2 = RandomForestClassifier(criterion='entropy', bootstrap = True, max_depth=4, min_samples_leaf=7)
clf_SMOTE2
        
clf_SMOTE2 = clf_SMOTE2.fit(X_train, Y_train)
    
    
a = df.columns[6:]
dot_data_2 = tree.plot_tree(clf_SMOTE2,  model.estimators_[5], feature_names= a)
graphviz.Source(dot_data_2)

推荐答案

来自help page：

随机森林是符合多个决策的元估计器 数据集不同子样本上的树分类器

因此您不能对RandomForestClassifier对象应用export_graphviz。您需要访问存储在estimators_：

下的其中一个诊断树

from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import make_classification
from sklearn.tree import export_graphviz
import pandas as pd

X,y = make_classification()
X = pd.DataFrame(X,columns = ['col'+str(i) for i in range(X.shape[1])])

clf = RandomForestClassifier(criterion='entropy', bootstrap = True, 
max_depth=4, min_samples_leaf=7)
clf = clf.fit(X, y)

在这里，我们只访问100个决策树中的第一个：

len(clf.estimators_)
100

dot_data_2 = export_graphviz(clf.estimators_[0], out_file=None, 
feature_names = X.columns, precision = 2, filled = False)

这篇关于导出/打印随机森林决策树/&#39；随机森林分类器对象没有属性&#39；树_&#39；的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！