File size: 906 Bytes
ed5e736 a5ecaaf d4546cc ed5e736 a5ecaaf d4546cc a5ecaaf ed5e736 a5ecaaf f7ea2e9 ed5e736 58b75ca ed5e736 f7ea2e9 ed5e736 f7ea2e9 ed5e736 | 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 26 27 28 29 30 31 32 33 | {
"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
} |