Graphviz 在 PyCharm 上不显示图片
我正在使用 sklearn 构建决策树模型。
我认为该模型工作得很好，但我不知道为什么它没有像 matplotlib 的 show() 函数那样自动显示图片。这和设定有关系吗？下面是代码：
import pandas as pd
from sklearn import tree
from sklearn.model_selection import train_test_split
from sklearn.tree import export_graphviz
import graphviz
data = pd.read_excel('file location')
target_vars = ['variable1','variable2','variable3']
X = pd.DataFrame()
for i in target_vars:
X[i]=data[i]
y = data['outcome']
X_tn, X_te, y_tn, y_te = train_test_split(X, y, random_state=0)
regr = tree.DecisionTreeClassifier(criterion='entropy', max_depth=5)
regr.fit(X_tn,y_tn)
y_pred = regr.predict(X_te)
accuracy = (y_pred==y_te).mean()
print('Model Accuracy: ', accuracy)
export_graphviz(regr, out_file='tree.dot', class_names=['1','0'],
feature_names=target_vars, impurity = True, filled = True)
with open('tree.dot') as f:
dot_graph = f.read()
graphviz.Source(dot_graph)
I am building a decision tree model using sklearn.
I think the model works just fine, but I have no idea why it is not showing the picture automatically, as matplotlib's show() function would. Does this have something to do with the setting? below is the code:
import pandas as pd
from sklearn import tree
from sklearn.model_selection import train_test_split
from sklearn.tree import export_graphviz
import graphviz
data = pd.read_excel('file location')
target_vars = ['variable1','variable2','variable3']
X = pd.DataFrame()
for i in target_vars:
X[i]=data[i]
y = data['outcome']
X_tn, X_te, y_tn, y_te = train_test_split(X, y, random_state=0)
regr = tree.DecisionTreeClassifier(criterion='entropy', max_depth=5)
regr.fit(X_tn,y_tn)
y_pred = regr.predict(X_te)
accuracy = (y_pred==y_te).mean()
print('Model Accuracy: ', accuracy)
export_graphviz(regr, out_file='tree.dot', class_names=['1','0'],
feature_names=target_vars, impurity = True, filled = True)
with open('tree.dot') as f:
dot_graph = f.read()
graphviz.Source(dot_graph)
如果你对这篇内容有疑问，欢迎到本站社区发帖提问 参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱，如果其他用户或者作者回复了您的评论，将不能在第一时间通知您！
发布评论
评论（1）
首先，您应该使用命令行在设备上安装 Graphviz，而不仅仅是在 PyCharm 中。
sudo apt-get install graphviz然后你可以像这样查看点文件：
first of all you should install
Graphvizon your device using command line not just in PyCharm.sudo apt-get install graphvizThen you can view the dot file like this: