File size: 535 Bytes
b83e315
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
# Конфигурация данных
data:
  train_path: "data/dataset_train.pt"
  test_path: "data/dataset_test.pt"
  batch_size: 32

# Параметры модели
model:
  bert_name: "cointegrated/rubert-tiny"

predct:
  use_model: "models/model_5.pth"
  threshold: 0.75

# Гиперпараметры обучения
training:
  epochs: 5
  learning_rate: 1e-5
  save_dir: "models"

# Настройки логирования
logging:
  log_dir: "logs"
  level: "INFO"
  format: "%(asctime)s - %(levelname)s - %(message)s"