我目前正在做一个关于假新闻检测的项目,并试图比较我使用的不同算法的准确性。现在我有了ROC曲线,如下所示:
到目前为止,我设法编写代码的是:
plt.figure(0).clf()
for model, name in [ (mn_count_clf, 'multinomial nb count'),
(mn_tfidf_clf, 'multinomial nb tfidf'),
(pa_tfidf_clf, 'passive aggressive'),
(svc_tfidf_clf, 'svc'),
(sgd_tfidf_clf, 'sgd')]:
if 'count' in name:
pred = model.predict_proba(count_test)[:,1]
elif 'multinomial' in name:
pred = model.predict_proba(tfidf_test)[:,1]
else:
pred = model.decision_function(tfidf_test)
fpr, tpr, thresh = metrics.roc_curve(y_test.values, pred, pos_label='REAL')
plt.plot(fpr,tpr,label="{}".format(name))
plt.legend(loc=0)
在python中,用于scikit-learn。曲线具有不同的颜色,这部分是我所希望的。
现在,为了清楚起见,我想要做的是改变不同情节的样式,即使用不同颜色的虚线,虚线和不同颜色的线。任何帮助表示赞赏。