在ROC图上更改曲线的颜色和样式

时间:2018-01-19 06:31:55

标签: python matplotlib scikit-learn roc

我目前正在做一个关于假新闻检测的项目,并试图比较我使用的不同算法的准确性。现在我有了ROC曲线,如下所示:

enter image description here

到目前为止,我设法编写代码的是:

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。曲线具有不同的颜色,这部分是我所希望的。

现在,为了清楚起见,我想要做的是改变不同情节的样式,即使用不同颜色的虚线,虚线和不同颜色的线。任何帮助表示赞赏。

0 个答案:

没有答案