TextClassifier / config.yaml
ivanovot
init
b83e315
raw
history blame contribute delete
535 Bytes
# Конфигурация данных
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"