| { | |
| "model_name": "bert-base-multilingual-cased", | |
| "model_id": "M4_4.1_mBERT", | |
| "member": 4, | |
| "spec_compliance": { | |
| "seed": 42, | |
| "max_length": 128, | |
| "batch_size": 16, | |
| "learning_rate": 2e-05, | |
| "epochs_trained": 3, | |
| "total_epochs_limit": 5, | |
| "early_stopping_patience": 5, | |
| "primary_metric": "val_macro_f1" | |
| }, | |
| "training_results": { | |
| "best_epoch": 3, | |
| "best_val_macro_f1": 0.6795928116428641, | |
| "best_val_macro_precision": 0.6946485613019967, | |
| "best_val_macro_recall": 0.6802384049141252, | |
| "best_val_weighted_f1": 0.7767666885575343, | |
| "best_val_accuracy": 0.7700119474313023, | |
| "training_loss": 0.6579384510566588 | |
| }, | |
| "test_results": { | |
| "test_loss": 0.925308403400838, | |
| "test_accuracy": 0.7541866028708134, | |
| "test_macro_precision": 0.6496175553843826, | |
| "test_macro_recall": 0.6441460137051936, | |
| "test_macro_f1": 0.6447829635587705, | |
| "test_weighted_precision": 0.7632761926177464, | |
| "test_weighted_recall": 0.7541866028708134, | |
| "test_weighted_f1": 0.7569916810574127, | |
| "per_class_f1": { | |
| "F": 0.8371837183718371, | |
| "R": 0.7635294117647059, | |
| "N": 0.8100526008182349, | |
| "M": 0.6145675265553869, | |
| "S": 0.19858156028368795 | |
| } | |
| }, | |
| "hardware": { | |
| "device": "cuda", | |
| "cuda_available": true, | |
| "python_version": "3.11.11", | |
| "platform": "Linux-6.8.0-87-generic-x86_64-with-glibc2.35", | |
| "cpu_count": 128, | |
| "ram_gb": 754.5698852539062, | |
| "gpu_device": "NVIDIA H100 NVL MIG 1g.24gb", | |
| "gpu_memory_gb": 23.219666944 | |
| }, | |
| "training_time": { | |
| "total_seconds": 1386.469988, | |
| "total_minutes": 23.107833133333333, | |
| "average_per_epoch": 277.29253600000004, | |
| "timestamp_start": "2025-12-25T10:13:31.474388", | |
| "timestamp_end": "2025-12-25T10:36:58.037200" | |
| }, | |
| "library_versions": { | |
| "pytorch": "2.7.1+cu118", | |
| "transformers": "4.57.3", | |
| "numpy": "1.26.4", | |
| "pandas": "2.2.3", | |
| "scikit-learn": "1.6.1", | |
| "python": "3.11.11" | |
| }, | |
| "data": { | |
| "train_samples": 15699, | |
| "val_samples": 3348, | |
| "test_samples": 3344, | |
| "num_classes": 5, | |
| "class_names": [ | |
| "F", | |
| "R", | |
| "N", | |
| "M", | |
| "S" | |
| ] | |
| } | |
| } |