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 制作自定义评分指标?

逻辑回归成本函数 scikit learn

如何评估 scikit learn LogisticRegression 的成本函数?

在 scikit learn 中自定义损失函数

Scikit-learn χ²(卡方)统计量和相应的列联表