Scikit.learn auc 函数类型错误

Scikit.learn auc function TypeError

我正在按照 计算数据中的 AUC 分数,但如果我尝试 运行 脚本,它会返回一个 TypeError。

TypeError: ufunc 'subtract' did not contain a loop with signature matching types dtype('S32') dtype('S32') dtype('S32')

我不明白,因为 roc_curve 函数工作正常,所以我希望 auc 函数也应该工作。 我已经检查了我的测试数组的dtype,它们都是float64

感谢您的帮助。

我发现了问题:

测试集和预测集的形状为 (N,),但函数 roc_curve 要求它们的形状为 (N,1)。 我解决了在数组上应用 Numpy.reshape 函数的问题。