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

来源:互联网 发布:文件同步备份软件 编辑:程序博客网 时间:2024/06/05 00:51

引用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)
阅读全文
0 0
原创粉丝点击