scikit-learn ExtraTreeClassifier 和 RandomForestClassifier 的图

Question

我正在尝试制作一些图表来说明 scikit-learn 中 RandomForestClassifier 和 ExtraTreeClassifier 之间的区别。我想我可能已经弄清楚了，但我不确定。这是我的代码来拟合和绘制虹膜数据集：

import numpy as np
from sklearn.datasets import load_iris
from sklearn.externals.six import StringIO 
from sklearn import tree
import pydot

iris = load_iris()

X = iris.data
y = iris.target

clf = tree.ExtraTreeClassifier()

clf = clf.fit(iris.data, iris.target)


dot_data = StringIO() 
tree.export_graphviz(clf, out_file=dot_data) 
graph = pydot.graph_from_dot_data(dot_data.getvalue()) 
file_name = "et_iris.pdf"
graph.write_pdf(file_name) 

clf = tree.DecisionTreeClassifier()

clf = clf.fit(iris.data, iris.target)


dot_data = StringIO() 
tree.export_graphviz(clf, out_file=dot_data) 
graph = pydot.graph_from_dot_data(dot_data.getvalue()) 
file_name = "rdf_iris.pdf"
graph.write_pdf(file_name) 

这产生的图看起来是正确的，ET 图比决策树图更“笨拙”。

DecisionTreeClassifier 与 RandomForestClassifier 中的一棵树相同，而 ExtraTreeClassifier 与 ExtraTreeClassifier 中的一棵树相同，我是否正确？

有没有办法对实际 RDF 或 ET 分类器中的所有树执行此操作？我尝试在森林中使用 .estimators_，但它们似乎没有导出方法。

Answer 1

export_graphviz 不是方法，而是函数。没有一棵树“拥有”它。您可以将其与estimators_ 一起使用。 关于ExtraTreeClassifier 是ExtraTreesClassifier 中的一棵树和DecisionTreeClassifier 是RandomForestClassifier 中的一棵树，你是对的。但是，这并没有真正涵盖它，因为：

• RandomForestClassifier 为每棵树单独引导数据集，ExtraTreesClassifier 不引导（默认情况下）。

• max_features=n_features默认用于单棵树，即所有特征都可以在每个拆分中使用。