決策樹是一個非參數的監督式學習方法,主要用於分類和迴歸。算法的目標是經過推斷數據特徵,學習決策規則從而建立一個預測目標變量的模型。html
from sklearn import tree X = [[0, 0], [1, 1]] Y = [0, 1] clf = tree.DecisionTreeClassifier() clf = clf.fit(X, Y) clf.predict([[2., 2.]]) clf.predict_proba([[2., 2.]]) #計算屬於每一個類的機率
要求根據要求隨機生成數據,並構建決策樹,並舉例預測。python
(1) 調用數據的方法以下:算法
from sklearn.datasets import load_iris iris = load_iris()# 從sklearn 數據集中獲取鳶尾花數據。
(2) 利用sklearn中的決策樹方法對鳶尾花數據創建決策樹
(3) 爲了可以直觀看到建好的決策樹,安裝 pydotplus, 方法以下:函數
pip install pydotplus
pydotplus使用方法工具
import pydotplus #引入pydotplus dot_data = tree.export_graphviz(clf, out_file=None) graph = pydotplus.graph_from_dot_data(dot_data) graph.write_pdf("iris.pdf")#將圖寫成pdf文件
(4)(選作) 不使用sklearn中的決策樹方法,本身編寫決策樹構建程序(建議用python語言),並對鳶尾花數據構建決策樹。學習
經過本次實驗,瞭解了Python中構建決策樹的函數方法,並用鳶尾花數據集的可視化看到了生成的決策樹效果圖。困難在於不太理解決策時具體的構建過程,通過屢次試驗,自動構建的決策樹和本身預測的規律結果是保持一致的,這說明了決策樹的實用性。下一步應該學習理解本身構建出決策樹。
原文出處:https://www.cnblogs.com/wonker/p/11062683.htmlspa