安裝graphviz
從以下鏈接下載GraphViz：https://www.php.cn/link/bfaa2ace1082af90074d02006690ddb2。
通過命令conda install python-graphviz來安裝graphviz的python庫。
生成可視化文件的代碼如下，使用JavaScript語言：
import graphviz // 忽略測試 from sklearn import tree print(data.columns) // ID3為決策樹分類器，fit之后得到的模型，注意必須在fit之后執行，predict之后運行會報錯 dot_data = tree.export_graphviz(ID3, out_file=None, feature_names=data.columns[:-1], class_names=np.unique(y)) // 忽略測試 graph = graphviz.Source(dot_data) // 忽略測試 // 在同級目錄下生成tree.pdf文件 graph.render("tree") // 忽略測試
立即學習“Python免費學習筆記（深入）”；
參考資料：https://www.php.cn/link/2cfc7c2a1eabaff9fdcc09b009e8edad
? 版權聲明
文章版權歸作者所有，未經允許請勿轉載。
THE END
喜歡就支持一下吧
相關推薦
