Rahilgh commited on
Commit
20c906a
·
verified ·
1 Parent(s): bce3031

Upload metadata.json

Browse files
Files changed (1) hide show
  1. metadata.json +79 -0
metadata.json ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "bert-base-multilingual-cased",
3
+ "model_id": "M4_4.1_mBERT",
4
+ "member": 4,
5
+ "spec_compliance": {
6
+ "seed": 42,
7
+ "max_length": 128,
8
+ "batch_size": 16,
9
+ "learning_rate": 2e-05,
10
+ "epochs_trained": 3,
11
+ "total_epochs_limit": 5,
12
+ "early_stopping_patience": 5,
13
+ "primary_metric": "val_macro_f1"
14
+ },
15
+ "training_results": {
16
+ "best_epoch": 3,
17
+ "best_val_macro_f1": 0.6795928116428641,
18
+ "best_val_macro_precision": 0.6946485613019967,
19
+ "best_val_macro_recall": 0.6802384049141252,
20
+ "best_val_weighted_f1": 0.7767666885575343,
21
+ "best_val_accuracy": 0.7700119474313023,
22
+ "training_loss": 0.6579384510566588
23
+ },
24
+ "test_results": {
25
+ "test_loss": 0.925308403400838,
26
+ "test_accuracy": 0.7541866028708134,
27
+ "test_macro_precision": 0.6496175553843826,
28
+ "test_macro_recall": 0.6441460137051936,
29
+ "test_macro_f1": 0.6447829635587705,
30
+ "test_weighted_precision": 0.7632761926177464,
31
+ "test_weighted_recall": 0.7541866028708134,
32
+ "test_weighted_f1": 0.7569916810574127,
33
+ "per_class_f1": {
34
+ "F": 0.8371837183718371,
35
+ "R": 0.7635294117647059,
36
+ "N": 0.8100526008182349,
37
+ "M": 0.6145675265553869,
38
+ "S": 0.19858156028368795
39
+ }
40
+ },
41
+ "hardware": {
42
+ "device": "cuda",
43
+ "cuda_available": true,
44
+ "python_version": "3.11.11",
45
+ "platform": "Linux-6.8.0-87-generic-x86_64-with-glibc2.35",
46
+ "cpu_count": 128,
47
+ "ram_gb": 754.5698852539062,
48
+ "gpu_device": "NVIDIA H100 NVL MIG 1g.24gb",
49
+ "gpu_memory_gb": 23.219666944
50
+ },
51
+ "training_time": {
52
+ "total_seconds": 1386.469988,
53
+ "total_minutes": 23.107833133333333,
54
+ "average_per_epoch": 277.29253600000004,
55
+ "timestamp_start": "2025-12-25T10:13:31.474388",
56
+ "timestamp_end": "2025-12-25T10:36:58.037200"
57
+ },
58
+ "library_versions": {
59
+ "pytorch": "2.7.1+cu118",
60
+ "transformers": "4.57.3",
61
+ "numpy": "1.26.4",
62
+ "pandas": "2.2.3",
63
+ "scikit-learn": "1.6.1",
64
+ "python": "3.11.11"
65
+ },
66
+ "data": {
67
+ "train_samples": 15699,
68
+ "val_samples": 3348,
69
+ "test_samples": 3344,
70
+ "num_classes": 5,
71
+ "class_names": [
72
+ "F",
73
+ "R",
74
+ "N",
75
+ "M",
76
+ "S"
77
+ ]
78
+ }
79
+ }