《机器学习》周志华 习题答案4.3

Posted 2020-08-02 机器人小z

　　原题，对西瓜数据集用决策树来进行划分，此处我只选取了西瓜的密度和含糖率这两个连续属性来进行划分，

#!/usr/bin/python
# -*- coding:utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors
from sklearn.discriminant_analysis import QuadraticDiscriminantAnalysis

file1 = open(\'c:\\quant\\watermelon.csv\',\'r\')
data = [line.strip(\'\\n\').split(\',\') for line in file1]
X = [[float(raw[-3]), float(raw[-2])] for raw in data[1:]]
y = [1 if raw[-1]==\'1\' else 0 for raw in data[1:]]
X = np.array(X)
y = np.array(y)
print y
#######################################################################以上是西瓜
from sklearn.datasets import load_iris
from sklearn import tree
from sklearn.externals.six import StringIO
import pydot
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)


#with open("iris.dot", \'w\') as f:
#    f = tree.export_graphviz(clf, out_file=f)
filename = "iris.pdf"
dot_data = StringIO()
tree.export_graphviz(clf, out_file=dot_data)

graph = pydot.graph_from_dot_data(dot_data.getvalue())
print graph
graph.write_pdf(filename)

结果如下：