Found ValueError:y_true中只有一个类,但Y变量同时具有二进制值

时间:2017-01-23 23:33:42

标签: python scikit-learn roc grid-search auc

我试图运行ROC auc,代码是:

actuals = gs.predict(feature_set) 
probas = gs.predict_proba(feature_set)
plt.plot(roc_curve(data[['Won_YES']], probas[:,1])[0],
                       roc_curve(data[['Won_YES']], probas[:,1])[1])

Won_YES是Y变量并且具有' 1'和' 0'值。它按值排序,因此我使用数据[' Won_YES'] .sample(frac = 0.1,replace = True)来应用随机顺序。所以不确定还有什么可以遗漏。

这是我得到的错误: ValueError:y_true中只有一个类。在这种情况下,没有定义ROC AUC分数。

0 个答案:

没有答案