基于iris数据集 实战sklearn决策树 利用graphviz可视化

python库安装：

• pip install graphviz
• pip install pydotplus 

graphviz安装：

• 下载地址：http://www.graphviz.org

完整代码实现

#!/usr/bin/python
# -*- coding: utf-8 -*-
# @author: T

from sklearn import datasets
from sklearn import model_selection
from sklearn import tree 
import pydotplus
import os
os.environ['PATH'] += os.pathsep + r'D:\graphviz-2\release\bin'

# 读取数据集
def readData():

	iris = datasets.load_iris()

	return iris.data, iris.target


def sklearn_tree(dataSet, target):
	# 数据划分
	train_x, test_x, train_y, test_y = model_selection.train_test_split(dataSet, target, test_size=0.1, random_state=0)
	# 定义决策树分类器 基尼指数
	clf = tree.DecisionTreeClassifier()
	# 训练分类器
	clf.fit(train_x, train_y)

	# 预测测试样本
	pre_y = cl