我正在按照这个StackOverflow的回答来计算我的数据的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
函数解决了这个问题。