category-e5-title / config.json
dbuyanov's picture
Train category classifier
d4546cc verified
Raw History Blame Contribute Delete
906 Bytes
{
"batch_size": 64,
"best_epoch": 11,
"best_validation_score": 0.7521795132006518,
"class_weight_mode": "inverse_sqrt",
"class_weight_summary": {
"max": 2.808645725250244,
"mean": 0.9999999403953552,
"min": 0.12191355228424072,
"mode": "inverse_sqrt"
},
"device": "cuda",
"early_stopping_metric": "validation_macro_f1",
"early_stopping_min_delta": 0.0001,
"early_stopping_patience": 2,
"elapsed_sec": 5307.876830555499,
"epochs": 12,
"eval_batch_size": 64,
"input_field": "embedding_text",
"input_path": "data/category/title/gpt-5.1-cleaned/dataset.filtered.csv",
"labels": 347,
"learning_rate": 2e-05,
"limit": null,
"max_length": 128,
"model_name": "intfloat/multilingual-e5-base",
"out_dir": "models/category-e5-title",
"seed": 42,
"test_rows": 11904,
"top_k": 5,
"train_rows": 95255,
"validation_rows": 11904,
"weight_decay": 0.01
}