机器学习练习——使用决策树实现iris分类器

import pydotplus
from sklearn.tree import DecisionTreeClassifier
from sklearn import datasets
from IPython.display import display, Image
from sklearn import tree

//使用官方数据集iris
iris = datasets.load_iris()
features = iris.data
target = iris.target
//构建分类器
decision_tree = DecisionTreeClassifier(random_state=0)
//训练模型
model = decision_tree.fit(features, target)
//可视化
dot_data = tree.export_graphviz(decision_tree,
                                out_file=None,
                                feature_names=iris.feature_names,
                                class_names=iris.target_names)

graph = pydotplus.graph_from_dot_data(dot_data)

graph.write_png('iris.png')