我想爲不同的分類器計算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曲線只能在二進制情況。它在多類設置中沒有意義。 –