scikit-learn 學習筆記 - Decision Trees

Decision Trees 官方文檔
Decision Trees 中文文檔

DecisionTreeClassifier 能夠二進制（其中標籤是[-1,1]）分類和多類（其中標籤是[0，…，K-1]）分類。
使用Iris數據集，我們可以構造一個樹，如下所示：

from sklearn.datasets import load_iris
from sklearn import tree
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)

import graphviz 
dot_data = tree.export_graphviz(clf, out_file=None) 
graph = graphviz.Source(dot_data) 
graph.render("iris") 

dot_data = tree.export_graphviz(clf, out_file=None, 
                         feature_names=iris.feature_names,  
                         class_names=iris.target_names,  
                         filled=True, rounded=True,  
                         special_characters=True)  
graph = graphviz.Source(dot_data)