Scikit.learn auc 函数 TypeError

Question

我正在关注 this StackOverflow answer 来计算我的数据中的 AUC 分数，但如果我尝试运行脚本，它会返回一个 TypeError。

TypeError: ufunc 'subtract' 不包含带有签名的循环 匹配类型 dtype('S32') dtype('S32') dtype('S32')

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

感谢您的帮助。

Answer 1

我发现了问题：

测试集和预测集具有形状 (N,)，但它们必须具有来自函数 roc_curve 的形状 (N,1)。 我解决了在数组上应用Numpy.reshape 函数的问题。