文章目錄
- 相應軟體安裝
- cmake安裝
- graphviz安裝
- 決策樹畫圖實操
相應軟體安裝
pip install graphviz
cmake安裝
下載地址
軟體安裝程序，要勾選添加環境變數：
然后一路下一步，就可以了~
graphviz安裝
安裝了Python庫graphviz，還不夠，還需要安裝graphviz下載頁面這個軟體，下載地址如下：軟體
軟體安裝程序，要勾選添加環境變數：
然后一路下一步，就可以了~
決策樹畫圖實操
常規匯出到圖片（沒有顏色設定）
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
iris = load_iris()
clf = DecisionTreeClassifier()
clf_iris = clf.fit(iris.data, iris.target)
import graphviz
# 將決策樹的結構資料匯出
dot_data = tree.export_graphviz(clf, out_file = './iris')
graph = graphviz.Source.from_file('./iris')
graph.render('iris',format='pdf')# 保存到檔案中
# 不匯出直接，使用
dot_data = tree.export_graphviz(clf, out_file = None)
graph = graphviz.Source(dot_data)
graph.render('iris',format='png')# 保存到檔案中
以下是，保存結果顯示：
pdf：
png：
更多設定匯出到圖片（彩色決策樹可視化圖片）
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
iris = load_iris()
clf = DecisionTreeClassifier()
clf_iris = clf.fit(iris.data, iris.target)
dot_data = tree.export_graphviz(clf, out_file=None,
feature_names=iris.feature_names,# 特征名
class_names=iris.target_names,# 類別名
filled=True, # 填充顏色
rounded=True) # 圓角
graph = graphviz.Source(dot_data)
graph.render('iris2') # 默認保存pdf格式
以下是，保存結果顯示效果：
我要做的事情是，庖丁解書，讓你看個通透，學個明白，既有文章，又有視頻，包教包會，學不會一定不是學生的問題，而是老師的問題！
我在馬士兵教育，教授AIoT-人工智能物聯網課程，歡迎你加入，深入學習~
代碼點亮人生，代碼改變世界~
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/289383.html
標籤：其他
上一篇：MS-RTOS --- 產品特性