From 89f188d9352e11d80a443e148730065e2f2b064e Mon Sep 17 00:00:00 2001
From: Link_pursuit <1149830513@qq.com>
Date: Fri, 29 Mar 2024 00:46:15 +0800
Subject: [PATCH] Update test_metric.py

---
 BERT/test_metric.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/BERT/test_metric.py b/BERT/test_metric.py
index ba2cee7..f1fb4dd 100644
--- a/BERT/test_metric.py
+++ b/BERT/test_metric.py
@@ -42,7 +42,7 @@ if __name__ == '__main__':
     data_loader = DataLoader(dataset, batch_size=batch_size, shuffle=False, num_workers=1)
 
     # 加载模型和检查点
-    model = BertForSequenceClassification.from_pretrained('./BERT_training_output/checkpoint-35544')
+    model = BertForSequenceClassification.from_pretrained('./BERT_training_output/checkpoint-35544'， num_labels = 6)
 
     # 运行推断
     model.eval()