cocoshe/bert-base-chinese-finetune-5-trash-email

Based on bert-base-chinese

基于bert-base-chinese在message80W数据集（垃圾邮件二分类）上做了5个epoch的fine-tune

# evaluate
with torch.no_grad():
    model.eval()
    eval_steps = 0
    pred_list = []
    label_list = []
    for i, batch in enumerate(tqdm(test_loader)):
        input_ids, attention_mask, label = batch
        logits = model(input_ids, attention_mask)
        pred_list += (torch.argmax(logits, dim=-1))
        label_list += label
        eval_steps += 1

80W数据，shuffled，8：3分train eval

下面是eval结果