PYTHON 決策樹可視化

Question

[![在此處輸入圖像描述][1]][1]我想可視化我已應用於 pdf 或 png 文件中的數據的樹決策分類器。 我嘗試通過以下代碼使用 graphviz 進行可視化：

X_train, X_test, y_train, y_test = \
        train_test_split(X, y, test_size=0.30, random_state=1)

clf =tree.DecisionTreeClassifier(max_depth=43)
clf = clf.fit(X_train, y_train)
from sklearn.externals.six import StringIO  
import pydot 
dot_data = StringIO() 
tree.export_graphviz(clf, out_file=dot_data) 
graph = pydot.graph_from_dot_data(dot_data.getvalue()) 
graph[0].write_pdf("tree.pdf") 

但程序無法完成。 一旦我收到內存不足的錯誤，第二次我收到錯誤“點停止工作”。 由於這個問題，我想通過知道左孩子在哪里，右孩子或左孩子在哪里來了解這棵樹？ 感謝您的任何回應和幫助

Answer 1

如果您收到類似以下內容的錯誤：

Program terminated with status: -11. stderr follows: dot: graph is too large for cairo-renderer bitmaps.

然后為了理解樹，您可以嘗試將其以樹文本格式顯示在屏幕上，如下所示：

from sklearn.tree import export_text

r = export_text(clf, feature_names=df_X_train.columns)
print(r)