Spaces:
Sleeping
Sleeping
# Конфигурация данных | |
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" |