告密者的下场

oss.backward()

optimizer.step()

_,pred=torch.max(outputs.data,dim=1)

acc=pred.eq(labels.data).cpu().sum()

one=torch.ones_like(labels)

zero=torch.zeros_like(labels)

tn=((labels==zero)*(pred==zero)).sum()

tp=((labels==one)*(pred==one)).sum()

fp=((labels==zero)*(pred==one)).sum()

fn=((labels==one)*(pred==zero)).sum()

train_sum_fn+=fn.item()

train_sum_fp+=fp.item()

train_sum_tn+=tn.item()

train_sum_tp+=tp.item()

train_sum_loss+=loss.item()

train_sum_correct+=acc.item()

（本章未完，请点击下一页继续阅读）

', ' ')('\n

告密者的下场 (第2/2页)

train_loss=train_sum_loss*1.0/len(trainDataLoader)

train_correct=train_sum_correct*1.0/len(trainDataLoader)/batch_size

train_precision=train_sum_tp*1.0/(train_sum_fp+train_sum_tp)

train_recall=train_sum_tp*1.0/(

本章未完，请点击下一页继续阅读！ 第5页 / 共7页