1

bert_train.py

@@ -10,7 +10,6 @@ from transformers import (
 import torch
 from torch.utils.data import Dataset
 from tqdm import tqdm
-import os
 import warnings
 import re  # 用于正则表达式清洗
 from sklearn.preprocessing import LabelEncoder

requirements.txt

@@ -25,6 +25,9 @@ Werkzeug==3.1.3
 pyyaml==6.0.2
 tqdm==4.67.1
 simplejson==3.20.1
+joblib==1.5.1
+typing_extensions==4.12.2
+typing-inspection==0.4.1
 
 
 torch==2.7.1  # 如果使用GPU或需要特定版本