aibanov / training_info.json
primel's picture
Upload training_info.json
b20e183 verified
{
"model_name": "google-bert/bert-base-multilingual-uncased",
"training_date": "2025-11-06 19:14:32",
"training_duration_minutes": 200.33368061383564,
"dataset_info": {
"total_samples": 566266,
"train": 481326,
"validation": 42470,
"test": 42470
},
"config": {
"dataset_path": "/content/drive/MyDrive/transaction_dataset.jsonl",
"drive_model_dir": "/content/drive/MyDrive/models/multitask_ner_intent_lang_v2",
"local_model_dir": "/content/multitask_model",
"model_name": "google-bert/bert-base-multilingual-uncased",
"max_length": 192,
"dropout": 0.15,
"num_epochs": 3,
"batch_size": 48,
"gradient_accumulation_steps": 2,
"learning_rate": 2e-05,
"weight_decay": 0.01,
"warmup_ratio": 0.05,
"max_grad_norm": 1.0,
"ner_weight": 0.5,
"intent_weight": 0.35,
"lang_weight": 0.15,
"eval_steps": 1000,
"save_steps": 1000,
"logging_steps": 200,
"save_to_drive_every_n_checkpoints": 3,
"filter_empty_entities": true,
"min_tokens": 2,
"max_samples_per_intent": null,
"seed": 42,
"test_size": 0.15,
"val_split": 0.5,
"use_fp16": true,
"dataloader_workers": 2
},
"final_results": {
"validation": {
"ner_f1": 0.9999503746712323,
"ner_precision": 0.999946239449492,
"ner_recall": 0.9999545099271745,
"intent_f1": 0.5543617045485377,
"intent_accuracy": 0.6771603484812809,
"lang_f1": 0.9186241343372614,
"lang_accuracy": 0.9434895220155404,
"avg_f1": 0.8243120711856772
},
"test": {
"ner_f1": 0.9998547862453532,
"ner_precision": 0.9998216004513923,
"ner_recall": 0.999887974242374,
"intent_f1": 0.5544819016678323,
"intent_accuracy": 0.6771838945137745,
"lang_f1": 0.9194956770328484,
"lang_accuracy": 0.9443842712502943,
"avg_f1": 0.8246107883153447
}
},
"label_counts": {
"num_ner_labels": 33,
"num_intents": 3,
"num_languages": 5
}
}