Scikit.learn auc 函数 TypeError
Posted
技术标签:
【中文标题】Scikit.learn auc 函数 TypeError【英文标题】:Scikit.learn auc function TypeError 【发布时间】:2017-08-28 03:56:58 【问题描述】:我正在关注 this *** answer 来计算我的数据中的 AUC 分数,但如果我尝试运行脚本,它会返回一个 TypeError。
TypeError: ufunc 'subtract' 不包含带有签名的循环 匹配类型 dtype('S32') dtype('S32') dtype('S32')
我不明白,因为roc_curve
函数工作正常,所以我希望auc
函数也应该工作。
我已经检查了我的测试数组的dtype
,它们都是float64
。
感谢您的帮助。
【问题讨论】:
【参考方案1】:我发现了问题:
测试集和预测集具有形状 (N,),但它们必须具有来自函数 roc_curve
的形状 (N,1)。
我解决了在数组上应用Numpy.reshape
函数的问题。
【讨论】:
以上是关于Scikit.learn auc 函数 TypeError的主要内容,如果未能解决你的问题,请参考以下文章
用于文本分类的 Scikit Learn-MultinomialNB
如何在 scikit learn 中为 cross_validate 制作自定义评分指标?