1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
| class Config:
| # 模型参数
| MODEL_NAME = "bert-base-chinese"
| MAX_LENGTH = 64
| BATCH_SIZE = 8
| EPOCHS = 3
| LEARNING_RATE = 2e-5
| WARMUP_STEPS = 500
| WEIGHT_DECAY = 0.01
|
| # 路径配置
| DATA_PATH = "data/sms_classify.csv"
| MODEL_PATH = "./models/classifier"
| LOG_PATH = "./logs"
|
| # 训练配置
| SEED = 42
| TEST_SIZE = 0.2
| EARLY_STOPPING_PATIENCE = 2
|
|
