From e8afd467faf3c2cd8c5f0689389fd4d260c989c2 Mon Sep 17 00:00:00 2001 From: renzhiyuan <465386466@qq.com> Date: Fri, 8 Aug 2025 15:23:31 +0800 Subject: [PATCH] 1 --- bert_train.py | 1 - requirements.txt | 3 +++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/bert_train.py b/bert_train.py index c5ecf6b..096d412 100644 --- a/bert_train.py +++ b/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 diff --git a/requirements.txt b/requirements.txt index 68988fd..bca03fd 100644 --- a/requirements.txt +++ b/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或需要特定版本