分类：训练模型输出

2021
07-01

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

1、计算F1-Score对于二分类来说，假设batchsize大小为64的话，那么模型一个batch的输出应该是torch.size([64,2])，所以首先做的是得到这个二维矩阵的每一行的最大索引值，然后添加到一个列表中，同时把标签也添加到一个列表中，最后使用sklearn中计算F1的工具包进行计算，代码如下importnumpyasnpimportsklearn.metricsimportf1_scoreprob_all=[]lable_all=[]fori,(data,label)intqdm(train_data_loader):prob... 继续阅读 >