我正在尝试渲染使用sklearn
的{{1}}创建的决策树对象。我正在使用这样的东西生成这棵树:
DecisionTreeClassifier
其中tree = DecisionTreeClassifier(max_depth=3, random_state=17).fit(X_train, y_train)
和X_train
分别是2D和1D数据帧。然后,我尝试在y_train
中渲染树,如下所示:
jupyter-notebook
我得到的不是输出渲染的图像,而是输出以下文本:
from sklearn.tree import export_graphviz
from io import StringIO
import pydotplus
from ipywidgets import Image
dot_data = StringIO()
export_graphviz(decision_tree=tree, feature_names=df.drop('Churn', axis=1).columns,
out_file=dot_data, filled=True)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
Image(value=graph.create_png())
为什么我得到这个而不是实际的图像?在this video的1:04:22似乎很好...