В Python sklearn.svm.OneClassSvm, как рассчитать ROC и AUC только с предсказанной меткой и реальной меткой?
Я использую Python 3.6,sklearn.svm.OneClassSVM для практики OSVM, и я хочу
рассчитать РПЦ, АУК.
Я использовал solution_function() для вычисления ROC и AUC, код приведен ниже.
Я хочу оценить значение, которое я рассчитываю по решению.
Могу ли я использовать только предсказанную метку и реальную метку для получения значения ROC, AUC?
y_score = oneclass.decision_function(testing_data)
roc_auc = metrics.roc_auc_score(Y_test, y_score)
1 ответ
Я не уверен, правильно ли я получил ваш вопрос, но если вы сделаете это:
clf = svm.OneClassSVM(nu=0.1, kernel="rbf", gamma=0.1)
clf.fit(X_train)
y_pred_train = clf.predict(X_train)
y_score = clf.predict(X_test)
Тогда вы сможете использовать:
from sklearn.metrics import roc_auc_score
roc_auc_score(y_test, y_score)
``