自觉机器猫

文章
9
资源
0
加入时间
2年10月21天

f1score python实现

f1_score(label,prediction)传入的两个参数都是一维的tensor,所以如果是多维的tensor,需要先将多维的展平(调用flatten()方法)为一维的tensor。注意:flatten()函数要求的参数类型是Tensor,不能是numpy里的array数组。所以如果拿到的数据是array数组,需要先将array数组转化成tensor。