| { |
| "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" |
| ] |
| } |
| } |
