Scikit.learn auc函数TypeError

Question

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

  

TypeError：ufunc'sundract'不包含带签名的循环   匹配类型dtype（'S32'）dtype（'S32'）dtype（'S32'）

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

感谢您的帮助。

Answer 1

我发现了问题：

测试和预测集具有形状（N，），但它们需要具有函数roc_curve的形状（N，1）。 我解决了在数组上应用Numpy.reshape函数的问题。