今天在使用tensorflow完成一个分类任务的时候,将输出的结果通过sklearn支持的一个classification_report这个函数来生成分类精度报告。然后出现了这样的错误。
ValueError: Classification metrics can\’t handle a mix of binary and continuous targets
在检查了predict和label的维数等之后,发现并不是维数的问题。
最终解决这个问题是发现predict和label的数据类型不同,predict还是float类型,表示分类类别的概率。label是int类型,代表的是类别标签。
然后把两者都统一为int之后问题解决。