我正在使用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()
###--------------------------------------------------------------------###
有什么帮助吗
保罗