f1score python实现_如何用python获取fscore指标

 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)