python随机森林可视化_使用Graphviz从Python中的随机森林模型可视化一棵树?

我正在使用Python(Pycharm社区版2016)

我用随机森林创建了一个工作模型,并且非常希望看到其中一棵树可视化。

我已经研究了很多关于Graphviz包是如何使用的信息,但是我还没有看到它的任何代码片段。

下面您可以看到一些创建模型的代码,我甚至使用preciion回调曲线来绘制它的精度。我现在如何包含代码来可视化其中一棵树?在###Split data into Training&Testing data-use stratified sampling ###

from sklearn.cross_validation import train_test_split

X = Actual_DataFrame.copy()

y = X.pop('Attrition')

X_train, X_test, y_train, y_test = train_test_split( X, y, test_size=0.30,

random_state=42, stratify=y)

###----------------------------------------------------------------------###

### Procedure:- (Model) - RF ###

from sklearn.ensemble import RandomForestClassifier

rf = RandomForestClassifier()

rf.fit(X_train, y_train)

predictions2 = rf.predict(X_test)

print ("(Model )")

print("Accuracy:",accuracy_score(y_test, predictions2))

print("Cohen's Kappa Score : ",cohen_kappa_score(y_test, predictions2))

print(classification_report(y_test, predictions2))

precision, recall, _ = precision_recall_curve(y_test, predictions2,

pos_label=1)

average_precision = average_precision_score(y_test, predictions2)

print (precision)

plt.plot(recall, precision, label='area = %0.2f' % average_precision,

color="green")

plt.xlim([0.0, 1.0])

plt.ylim([0.0, 1.05])

plt.xlabel('Recall')

plt.ylabel('Precision')

plt.title('Precision Recall Curve - RF')

plt.legend(loc="lower right")

plt.show()

###--------------------------------------------------------------------###

有什么帮助吗

保罗

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值