代碼

# -*- coding: utf-8 -*- """ Created on Mon Nov 25 19:54:07 2019@author: 劉云生 @blog:https://blog.csdn.net/liuyunshengsir 加返利機器人：lys20191020 """from sklearn.datasets import load_iris from sklearn import tree iris = load_iris() clf = tree.DecisionTreeClassifier() clf = clf.fit(iris.data, iris.target)import pandas as pddata=pd.DataFrame(iris.data,columns=['sepal length (cm)','sepal width (cm)','petal length (cm)','petal width (cm)'])label=pd.DataFrame(iris.target ,columns=['target'])import graphviz import os os.environ["PATH"] += os.pathsep + 'C:\\Program Files (x86)\\Graphviz2.38\\bin'dot_data = tree.export_graphviz(clf, out_file=None) graph = graphviz.Source(dot_data) graph.render("iris") graph.render(filename ="iris", directory ='./', format='pdf')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) print(graph)

實驗結果

總結

以上是生活随笔為你收集整理的决策树《机器学习入门学习》的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。