您的位置:首页 > 产品设计 > UI/UE

F1-Score "ValueError: Can't handle mix of binary and continuous"

2017-12-09 21:09 441 查看
引用StackOverflow:这里

其实很简单:比如我的真实数据标签是:

array([1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 0, 0])

下面是用回归得到的样本得分,注意这是回归得到的分数:

array([ 0.07094605, 0.1994941 , 0.19270157, 0.13379635, 0.04654469,

0.09212494, 0.19952108, 0.12884365, 0.15685076, -0.01274453,

0.32167554, 0.32167554, -0.10023553, 0.09819648, -0.06755516,

0.25390082, 0.17248324])

因为FI-score,accuracy_score这类函数是对二分类问题使用的得分,直接用API会报错:

accuracy_score(y_true, y_pred, normalize=False)


解决方法是使用阈值:

y_pre = [int(item>threshold) for  item in y_pred]
accuracy_score(y_true, y_pre , normalize=False)
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐