Upload metadata.json
Browse files- metadata.json +90 -0
metadata.json
ADDED
|
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "xlm-roberta-large",
|
| 3 |
+
"model_id": "M4_4.2_XLM_RoBERTa",
|
| 4 |
+
"member": 4,
|
| 5 |
+
"spec_compliance": {
|
| 6 |
+
"seed": 42,
|
| 7 |
+
"max_length": 128,
|
| 8 |
+
"batch_size": 8,
|
| 9 |
+
"gradient_accumulation_steps": 2,
|
| 10 |
+
"effective_batch_size": 16,
|
| 11 |
+
"learning_rate": 1e-05,
|
| 12 |
+
"epochs_trained": 3,
|
| 13 |
+
"total_epochs_limit": 5,
|
| 14 |
+
"early_stopping_patience": 5,
|
| 15 |
+
"primary_metric": "val_macro_f1",
|
| 16 |
+
"data_augmentation": true,
|
| 17 |
+
"target_minority_size": 0.05
|
| 18 |
+
},
|
| 19 |
+
"training_results": {
|
| 20 |
+
"best_epoch": 3,
|
| 21 |
+
"best_val_macro_f1": 0.6920319606014477,
|
| 22 |
+
"best_val_macro_precision": 0.6935400158042497,
|
| 23 |
+
"best_val_macro_recall": 0.6959520837420035,
|
| 24 |
+
"best_val_weighted_f1": 0.7889787090902882,
|
| 25 |
+
"best_val_accuracy": 0.7843488649940262,
|
| 26 |
+
"training_loss": 0.531401620691845
|
| 27 |
+
},
|
| 28 |
+
"test_results": {
|
| 29 |
+
"test_loss": 0.8419096413437215,
|
| 30 |
+
"test_accuracy": 0.7831937799043063,
|
| 31 |
+
"test_macro_precision": 0.7109259717505931,
|
| 32 |
+
"test_macro_recall": 0.6740476446561174,
|
| 33 |
+
"test_macro_f1": 0.682211638723438,
|
| 34 |
+
"test_weighted_precision": 0.7917048623914406,
|
| 35 |
+
"test_weighted_recall": 0.7831937799043063,
|
| 36 |
+
"test_weighted_f1": 0.78430327538584,
|
| 37 |
+
"per_class_f1": {
|
| 38 |
+
"F": 0.8503589177250138,
|
| 39 |
+
"R": 0.804436660828955,
|
| 40 |
+
"N": 0.8322618351841029,
|
| 41 |
+
"M": 0.6457399103139013,
|
| 42 |
+
"S": 0.2782608695652174
|
| 43 |
+
}
|
| 44 |
+
},
|
| 45 |
+
"data_augmentation": {
|
| 46 |
+
"original_train_size": 15699,
|
| 47 |
+
"augmented_train_size": 16156,
|
| 48 |
+
"augmented_samples": 457,
|
| 49 |
+
"target_minority_percentage": 5.0
|
| 50 |
+
},
|
| 51 |
+
"hardware": {
|
| 52 |
+
"device": "cuda",
|
| 53 |
+
"cuda_available": true,
|
| 54 |
+
"python_version": "3.11.11",
|
| 55 |
+
"platform": "Linux-6.8.0-87-generic-x86_64-with-glibc2.35",
|
| 56 |
+
"cpu_count": 128,
|
| 57 |
+
"ram_gb": 754.5698852539062,
|
| 58 |
+
"gpu_device": "NVIDIA H100 NVL MIG 1g.24gb",
|
| 59 |
+
"gpu_memory_gb": 23.219666944
|
| 60 |
+
},
|
| 61 |
+
"training_time": {
|
| 62 |
+
"total_seconds": 4912.679862,
|
| 63 |
+
"total_minutes": 81.8779977,
|
| 64 |
+
"average_per_epoch": 982.5344182000001,
|
| 65 |
+
"timestamp_start": "2025-12-25T13:51:24.634125",
|
| 66 |
+
"timestamp_end": "2025-12-25T16:18:05.909736"
|
| 67 |
+
},
|
| 68 |
+
"library_versions": {
|
| 69 |
+
"pytorch": "2.7.1+cu118",
|
| 70 |
+
"transformers": "4.57.3",
|
| 71 |
+
"numpy": "1.26.4",
|
| 72 |
+
"pandas": "2.2.3",
|
| 73 |
+
"scikit-learn": "1.6.1",
|
| 74 |
+
"python": "3.11.11"
|
| 75 |
+
},
|
| 76 |
+
"data": {
|
| 77 |
+
"train_samples_original": 15699,
|
| 78 |
+
"train_samples_augmented": 16156,
|
| 79 |
+
"val_samples": 3348,
|
| 80 |
+
"test_samples": 3344,
|
| 81 |
+
"num_classes": 5,
|
| 82 |
+
"class_names": [
|
| 83 |
+
"F",
|
| 84 |
+
"R",
|
| 85 |
+
"N",
|
| 86 |
+
"M",
|
| 87 |
+
"S"
|
| 88 |
+
]
|
| 89 |
+
}
|
| 90 |
+
}
|