安裝graphviz
從以下鏈接下載GraphViz：https://www.php.cn/link/bfaa2ace1082af90074d02006690ddb2。
通過(guò)命令conda install python-graphviz來(lái)安裝graphviz的Python庫(kù)。
生成可視化文件的代碼如下，使用JavaScript語(yǔ)言：
import graphviz // 忽略測(cè)試 from sklearn import tree print(data.columns) // ID3為決策樹(shù)分類器，fit之后得到的模型，注意必須在fit之后執(zhí)行，predict之后運(yùn)行會(huì)報(bào)錯(cuò) dot_data = tree.export_graphviz(ID3, out_file=None, feature_names=data.columns[:-1], class_names=np.unique(y)) // 忽略測(cè)試 graph = graphviz.Source(dot_data) // 忽略測(cè)試 // 在同級(jí)目錄下生成tree.pdf文件 graph.render("tree") // 忽略測(cè)試
立即學(xué)習(xí)“Python免費(fèi)學(xué)習(xí)筆記（深入）”；
參考資料：https://www.php.cn/link/2cfc7c2a1eabaff9fdcc09b009e8edad
.png)
推廣.jpg)