概述
f1_score(label,prediction)传入的两个参数都是一维的tensor，所以如果是多维的tensor，需要先将多维的展平（调用flatten()方法）为一维的tensor
注意：flatten()函数要求的参数类型是Tensor，不能是numpy里的array数组
所以如果拿到的数据是array数组，需要先将array数组转化成tensor
TypeError: flatten(): argument 'input' (position 1) must be Tensor, not numpy.ndarray
label_tensor=torch.from_numpy(label_array)
prediction_tensor=torch.from_numpy(prediction_array)
label_final=torch.flatten(label_tensor)
prediction_final=torch.flatten(prediction_tensor)
fl_score(label_final,prediction_final)
最后
以上就是自觉机器猫为你收集整理的f1score python实现的全部内容，希望文章能够帮你解决f1score python实现所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复