python - ROC 曲线和 AUC

Question

我想使用下面的代码计算 RandomForest 模型的 (y_test, y_pred) 的 AUC。但是有一些错误。它说 ValueError：不支持多类格式。看起来像这样：错误详细信息。我该如何解决，伙计们？非常感谢！

from sklearn.metrics import roc_curve, auc
false_positive_rate, true_positive_rate, thresholds = roc_curve(y_test, y_pred)
roc_auc = auc(false_positive_rate, true_positive_rate)
roc_auc

标签： pythonrandom-forestauc