Views
No views yet
message80W数据集（垃圾邮件二分类）上做了5个epoch的fine-tune1# evaluate
2with torch.no_grad():
3 model.eval()
4 eval_steps = 0
5 pred_list = []
6 label_list = []
7 for i, batch in enumerate(tqdm(test_loader)):
8 input_ids, attention_mask, label = batch
9 logits = model(input_ids, attention_mask)
10 pred_list += (torch.argmax(logits, dim=-1))
11 label_list += label
12 eval_steps += 1