我遵循教程來顯示roc曲線和相應的auc;我從來沒有使用ggplot庫,因此我不明白我的錯誤在哪裏。這裏下面的代碼:Python,Roc曲線和ggplot?
from sklearn import metrics
import pandas as pd
from ggplot import *
preds = clf.predict_proba(Xtest)[:,1]
fpr, tpr, _ = metrics.roc_curve(ytest, preds)
df = pd.DataFrame(dict(fpr=fpr, tpr=tpr))
ggplot(df, aes(x='fpr', y='tpr')) + geom_line() + geom_abline(linetype='dashed')
這是錯誤:
slope needed for <ggplot.geoms.geom_abline.geom_abline object at 0x7fae7e7f8d90>
我怎麼能解決這個問題?
你真的需要使用ggplot庫嗎?請參閱http://scikit-learn.org/stable/auto_examples/model_selection/plot_roc_crossval.html或http://scikit-learn.org/stable/auto_examples/model_selection/plot_roc.html查找matplotlib示例 – dukebody