Spaces:
Sleeping
Sleeping
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" |