我想为不同的分类器计算roc_auc
。有些不是二进制分类器。下面是我使用的代码的一部分:在scikit-learn中没有二进制分类器的auc得分
if hasattr(clf, "decision_function"):
y_score = clf.fit(X_train, y_train).decision_function(X_test)
else:
y_score = clf.fit(X_train, y_train).predict_proba(X_test)
AUC=roc_auc_score(y_test, y_score)
不过,我得到了一些分类错误(例如近邻 ):
ValueError: bad input shape
短短一句话,我用:y_score = clf.fit(X_train, y_train).predict_proba(X_test)
,但我真的不知道使用它是否正确。第一
clf.fit(X_train, y_train)
ROC曲线只能在二进制情况。它在多类设置中没有意义。 –