网站首页 训练模型

• 

  Pytorch训练模型得到输出后计算F1-Score 和AUC的操作

  1、计算F1-Score对于二分类来说，假设batchsize大小为64的话，那么模型一个batch的输出应该是torch.size([64,2])，所以首先做的是得到这个二维矩阵的每一行的最大索引值，然后添加到一个列表中，同时把标签也...

  2021-07-12 longtao100 python 521 ℃ 0 评论