parlange commited on
Commit
8b0b84e
·
verified ·
1 Parent(s): 38dd8f1

Upload PiT model from experiment a2

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +2 -0
  2. README.md +161 -0
  3. config.json +76 -0
  4. confusion_matrices/PiT_Confusion_Matrix_a.png +0 -0
  5. confusion_matrices/PiT_Confusion_Matrix_b.png +0 -0
  6. confusion_matrices/PiT_Confusion_Matrix_c.png +0 -0
  7. confusion_matrices/PiT_Confusion_Matrix_d.png +0 -0
  8. confusion_matrices/PiT_Confusion_Matrix_e.png +0 -0
  9. confusion_matrices/PiT_Confusion_Matrix_f.png +0 -0
  10. confusion_matrices/PiT_Confusion_Matrix_g.png +0 -0
  11. confusion_matrices/PiT_Confusion_Matrix_h.png +0 -0
  12. confusion_matrices/PiT_Confusion_Matrix_i.png +0 -0
  13. confusion_matrices/PiT_Confusion_Matrix_j.png +0 -0
  14. confusion_matrices/PiT_Confusion_Matrix_k.png +0 -0
  15. confusion_matrices/PiT_Confusion_Matrix_l.png +0 -0
  16. evaluation_results.csv +133 -0
  17. model.safetensors +3 -0
  18. pit-gravit-a2.pth +3 -0
  19. pytorch_model.bin +3 -0
  20. roc_confusion_matrix/PiT_roc_confusion_matrix_a.png +0 -0
  21. roc_confusion_matrix/PiT_roc_confusion_matrix_b.png +0 -0
  22. roc_confusion_matrix/PiT_roc_confusion_matrix_c.png +0 -0
  23. roc_confusion_matrix/PiT_roc_confusion_matrix_d.png +0 -0
  24. roc_confusion_matrix/PiT_roc_confusion_matrix_e.png +0 -0
  25. roc_confusion_matrix/PiT_roc_confusion_matrix_f.png +0 -0
  26. roc_confusion_matrix/PiT_roc_confusion_matrix_g.png +0 -0
  27. roc_confusion_matrix/PiT_roc_confusion_matrix_h.png +0 -0
  28. roc_confusion_matrix/PiT_roc_confusion_matrix_i.png +0 -0
  29. roc_confusion_matrix/PiT_roc_confusion_matrix_j.png +0 -0
  30. roc_confusion_matrix/PiT_roc_confusion_matrix_k.png +0 -0
  31. roc_confusion_matrix/PiT_roc_confusion_matrix_l.png +0 -0
  32. roc_curves/PiT_ROC_a.png +0 -0
  33. roc_curves/PiT_ROC_b.png +0 -0
  34. roc_curves/PiT_ROC_c.png +0 -0
  35. roc_curves/PiT_ROC_d.png +0 -0
  36. roc_curves/PiT_ROC_e.png +0 -0
  37. roc_curves/PiT_ROC_f.png +0 -0
  38. roc_curves/PiT_ROC_g.png +0 -0
  39. roc_curves/PiT_ROC_h.png +0 -0
  40. roc_curves/PiT_ROC_i.png +0 -0
  41. roc_curves/PiT_ROC_j.png +0 -0
  42. roc_curves/PiT_ROC_k.png +0 -0
  43. roc_curves/PiT_ROC_l.png +0 -0
  44. training_curves/PiT_accuracy.png +0 -0
  45. training_curves/PiT_auc.png +0 -0
  46. training_curves/PiT_combined_metrics.png +3 -0
  47. training_curves/PiT_f1.png +0 -0
  48. training_curves/PiT_loss.png +0 -0
  49. training_curves/PiT_metrics.csv +95 -0
  50. training_metrics.csv +95 -0
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ training_curves/PiT_combined_metrics.png filter=lfs diff=lfs merge=lfs -text
37
+ training_notebook_a2.ipynb filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,161 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - vision-transformer
5
+ - image-classification
6
+ - pytorch
7
+ - timm
8
+ - pit
9
+ - gravitational-lensing
10
+ - strong-lensing
11
+ - astronomy
12
+ - astrophysics
13
+ datasets:
14
+ - C21
15
+ metrics:
16
+ - accuracy
17
+ - auc
18
+ - f1
19
+ model-index:
20
+ - name: PiT-a2
21
+ results:
22
+ - task:
23
+ type: image-classification
24
+ name: Strong Gravitational Lens Discovery
25
+ dataset:
26
+ type: common-test-sample
27
+ name: Common Test Sample (More et al. 2024)
28
+ metrics:
29
+ - type: accuracy
30
+ value: 0.7813
31
+ name: Average Accuracy
32
+ - type: auc
33
+ value: 0.8358
34
+ name: Average AUC-ROC
35
+ - type: f1
36
+ value: 0.4987
37
+ name: Average F1-Score
38
+ ---
39
+
40
+ # 🌌 pit-gravit-a2
41
+
42
+ 🔭 This model is part of **GraViT**: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery
43
+
44
+ 🔗 **GitHub Repository**: [https://github.com/parlange/gravit](https://github.com/parlange/gravit)
45
+
46
+ ## 🛰️ Model Details
47
+
48
+ - **🤖 Model Type**: PiT
49
+ - **🧪 Experiment**: A2 - C21-half
50
+ - **🌌 Dataset**: C21
51
+ - **🪐 Fine-tuning Strategy**: half
52
+
53
+
54
+
55
+ ## 💻 Quick Start
56
+
57
+ ```python
58
+ import torch
59
+ import timm
60
+
61
+ # Load the model directly from the Hub
62
+ model = timm.create_model(
63
+ 'hf-hub:parlange/pit-gravit-a2',
64
+ pretrained=True
65
+ )
66
+ model.eval()
67
+
68
+ # Example inference
69
+ dummy_input = torch.randn(1, 3, 224, 224)
70
+ with torch.no_grad():
71
+ output = model(dummy_input)
72
+ predictions = torch.softmax(output, dim=1)
73
+ print(f"Lens probability: {predictions[0][1]:.4f}")
74
+ ```
75
+
76
+ ## ⚡️ Training Configuration
77
+
78
+ **Training Dataset:** C21 (Cañameras et al. 2021)
79
+ **Fine-tuning Strategy:** half
80
+
81
+
82
+ | 🔧 Parameter | 📝 Value |
83
+ |--------------|----------|
84
+ | Batch Size | 192 |
85
+ | Learning Rate | AdamW with ReduceLROnPlateau |
86
+ | Epochs | 100 |
87
+ | Patience | 10 |
88
+ | Optimizer | AdamW |
89
+ | Scheduler | ReduceLROnPlateau |
90
+ | Image Size | 224x224 |
91
+ | Fine Tune Mode | half |
92
+ | Stochastic Depth Probability | 0.1 |
93
+
94
+
95
+ ## 📈 Training Curves
96
+
97
+ ![Combined Training Metrics](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/training_curves/PiT_combined_metrics.png)
98
+
99
+
100
+ ## 🏁 Final Epoch Training Metrics
101
+
102
+ | Metric | Training | Validation |
103
+ |:---------:|:-----------:|:-------------:|
104
+ | 📉 Loss | 0.1100 | 0.0915 |
105
+ | 🎯 Accuracy | 0.9553 | 0.9710 |
106
+ | 📊 AUC-ROC | 0.9927 | 0.9943 |
107
+ | ⚖️ F1 Score | 0.9553 | 0.9710 |
108
+
109
+
110
+ ## ☑️ Evaluation Results
111
+
112
+ ### ROC Curves and Confusion Matrices
113
+
114
+ Performance across all test datasets (a through l) in the Common Test Sample (More et al. 2024):
115
+
116
+ ![ROC + Confusion Matrix - Dataset A](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_a.png)
117
+ ![ROC + Confusion Matrix - Dataset B](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_b.png)
118
+ ![ROC + Confusion Matrix - Dataset C](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_c.png)
119
+ ![ROC + Confusion Matrix - Dataset D](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_d.png)
120
+ ![ROC + Confusion Matrix - Dataset E](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_e.png)
121
+ ![ROC + Confusion Matrix - Dataset F](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_f.png)
122
+ ![ROC + Confusion Matrix - Dataset G](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_g.png)
123
+ ![ROC + Confusion Matrix - Dataset H](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_h.png)
124
+ ![ROC + Confusion Matrix - Dataset I](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_i.png)
125
+ ![ROC + Confusion Matrix - Dataset J](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_j.png)
126
+ ![ROC + Confusion Matrix - Dataset K](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_k.png)
127
+ ![ROC + Confusion Matrix - Dataset L](https://huggingface.co/parlange/pit-gravit-a2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_l.png)
128
+
129
+ ### 📋 Performance Summary
130
+
131
+ Average performance across 12 test datasets from the Common Test Sample (More et al. 2024):
132
+
133
+ | Metric | Value |
134
+ |-----------|----------|
135
+ | 🎯 Average Accuracy | 0.7813 |
136
+ | 📈 Average AUC-ROC | 0.8358 |
137
+ | ⚖️ Average F1-Score | 0.4987 |
138
+
139
+
140
+ ## 📘 Citation
141
+
142
+ If you use this model in your research, please cite:
143
+
144
+ ```bibtex
145
+ @misc{parlange2025gravit,
146
+ title={GraViT: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery},
147
+ author={René Parlange and Juan C. Cuevas-Tello and Octavio Valenzuela and Omar de J. Cabrera-Rosas and Tomás Verdugo and Anupreeta More and Anton T. Jaelani},
148
+ year={2025},
149
+ eprint={2509.00226},
150
+ archivePrefix={arXiv},
151
+ primaryClass={cs.CV},
152
+ url={https://arxiv.org/abs/2509.00226},
153
+ }
154
+ ```
155
+
156
+ ---
157
+
158
+
159
+ ## Model Card Contact
160
+
161
+ For questions about this model, please contact the author through: https://github.com/parlange/
config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "vit_base_patch16_224",
3
+ "num_classes": 2,
4
+ "num_features": 1000,
5
+ "global_pool": "avg",
6
+ "crop_pct": 0.875,
7
+ "interpolation": "bicubic",
8
+ "mean": [
9
+ 0.485,
10
+ 0.456,
11
+ 0.406
12
+ ],
13
+ "std": [
14
+ 0.229,
15
+ 0.224,
16
+ 0.225
17
+ ],
18
+ "first_conv": "conv1",
19
+ "classifier": "fc",
20
+ "input_size": [
21
+ 3,
22
+ 224,
23
+ 224
24
+ ],
25
+ "pool_size": [
26
+ 7,
27
+ 7
28
+ ],
29
+ "pretrained_cfg": {
30
+ "tag": "gravit_a2",
31
+ "custom_load": false,
32
+ "input_size": [
33
+ 3,
34
+ 224,
35
+ 224
36
+ ],
37
+ "fixed_input_size": true,
38
+ "interpolation": "bicubic",
39
+ "crop_pct": 0.875,
40
+ "crop_mode": "center",
41
+ "mean": [
42
+ 0.485,
43
+ 0.456,
44
+ 0.406
45
+ ],
46
+ "std": [
47
+ 0.229,
48
+ 0.224,
49
+ 0.225
50
+ ],
51
+ "num_classes": 2,
52
+ "pool_size": [
53
+ 7,
54
+ 7
55
+ ],
56
+ "first_conv": "conv1",
57
+ "classifier": "fc"
58
+ },
59
+ "model_name": "pit_gravit_a2",
60
+ "experiment": "a2",
61
+ "training_strategy": "half",
62
+ "dataset": "C21",
63
+ "hyperparameters": {
64
+ "batch_size": "192",
65
+ "learning_rate": "AdamW with ReduceLROnPlateau",
66
+ "epochs": "100",
67
+ "patience": "10",
68
+ "optimizer": "AdamW",
69
+ "scheduler": "ReduceLROnPlateau",
70
+ "image_size": "224x224",
71
+ "fine_tune_mode": "half",
72
+ "stochastic_depth_probability": "0.1"
73
+ },
74
+ "hf_hub_id": "parlange/pit-gravit-a2",
75
+ "license": "apache-2.0"
76
+ }
confusion_matrices/PiT_Confusion_Matrix_a.png ADDED
confusion_matrices/PiT_Confusion_Matrix_b.png ADDED
confusion_matrices/PiT_Confusion_Matrix_c.png ADDED
confusion_matrices/PiT_Confusion_Matrix_d.png ADDED
confusion_matrices/PiT_Confusion_Matrix_e.png ADDED
confusion_matrices/PiT_Confusion_Matrix_f.png ADDED
confusion_matrices/PiT_Confusion_Matrix_g.png ADDED
confusion_matrices/PiT_Confusion_Matrix_h.png ADDED
confusion_matrices/PiT_Confusion_Matrix_i.png ADDED
confusion_matrices/PiT_Confusion_Matrix_j.png ADDED
confusion_matrices/PiT_Confusion_Matrix_k.png ADDED
confusion_matrices/PiT_Confusion_Matrix_l.png ADDED
evaluation_results.csv ADDED
@@ -0,0 +1,133 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Model,Dataset,Loss,Accuracy,AUCROC,F1
2
+ ViT,a,0.3987342550652614,0.8997170701037409,0.917220073664825,0.46921797004991683
3
+ ViT,b,0.37321944226074877,0.9182646966362779,0.9273655616942909,0.5202952029520295
4
+ ViT,c,0.821146962441052,0.7922037095253065,0.866000920810313,0.2990455991516437
5
+ ViT,d,0.223954888615208,0.9440427538509902,0.9518121546961327,0.6130434782608696
6
+ ViT,e,1.1019757730900652,0.7782656421514819,0.85117687126315,0.5826446280991735
7
+ ViT,f,0.43205039906387277,0.8869181318255751,0.9119053612426381,0.1618828932261768
8
+ ViT,g,0.1527516215024516,0.9611666666666666,0.9983996666666667,0.962461736748832
9
+ ViT,h,0.39022785605769605,0.8943333333333333,0.9959461111111112,0.9040556900726392
10
+ ViT,i,0.07361652948241681,0.9748333333333333,0.9992922222222222,0.9753469387755102
11
+ ViT,j,6.027309565424919,0.5033333333333333,0.49077061111111114,0.13872832369942195
12
+ ViT,k,5.948174474835396,0.517,0.5691439444444444,0.14209591474245115
13
+ ViT,l,2.1620761937842525,0.7761620221035376,0.7346207805818022,0.6140942656577628
14
+ MLP-Mixer,a,1.0354112619425808,0.7239861678717384,0.9444742173112339,0.2779605263157895
15
+ MLP-Mixer,b,0.882929647823281,0.7834014460861365,0.9520524861878453,0.3291139240506329
16
+ MLP-Mixer,c,1.6300886590238712,0.6205595724614901,0.9170009208103129,0.21877022653721684
17
+ MLP-Mixer,d,0.055560619117974934,0.9792518076076705,0.9959686924493554,0.8366336633663366
18
+ MLP-Mixer,e,1.1025473987755214,0.70801317233809,0.9327480511617346,0.5596026490066225
19
+ MLP-Mixer,f,0.9539809344458585,0.763147703508636,0.9512486274645963,0.09952885747938751
20
+ MLP-Mixer,g,0.4660766951590776,0.8855,0.9941798888888888,0.8969551522423879
21
+ MLP-Mixer,h,0.8621955468207598,0.7991666666666667,0.9892695555555555,0.8322894919972165
22
+ MLP-Mixer,i,0.027433219969272612,0.9893333333333333,0.9997324444444444,0.9894109861019192
23
+ MLP-Mixer,j,5.562473163604737,0.4046666666666667,0.2374668888888889,0.05552617662612375
24
+ MLP-Mixer,k,5.123829672321677,0.5085,0.47802688888888895,0.06647673314339982
25
+ MLP-Mixer,l,2.271000012816855,0.6846808735656497,0.6265659288601144,0.5226162837242815
26
+ CvT,a,0.7062503800231756,0.6988368437598239,0.8373821362799264,0.2336
27
+ CvT,b,0.8609082461227902,0.6444514303678088,0.8087320441988951,0.20520028109627547
28
+ CvT,c,0.8150388154171053,0.6516818610499843,0.8144069981583795,0.20857142857142857
29
+ CvT,d,0.046723183949472995,0.9833385727758567,0.9917753222836095,0.8463768115942029
30
+ CvT,e,1.055778265130245,0.5916575192096597,0.7447665178233557,0.4397590361445783
31
+ CvT,f,0.6479923738885578,0.7303074897374332,0.8562897926766284,0.07737148913619502
32
+ CvT,g,0.47873973870277403,0.8053333333333333,0.948009388888889,0.8337129840546698
33
+ CvT,h,0.45442124152183533,0.8091666666666667,0.9536063888888889,0.8364519354377946
34
+ CvT,i,0.0470859190672636,0.985,0.999245111111111,0.9848637739656912
35
+ CvT,j,3.978341913700104,0.31966666666666665,0.09071466666666667,0.006812652068126521
36
+ CvT,k,3.5466880963295697,0.49933333333333335,0.5394476666666667,0.009234828496042216
37
+ CvT,l,1.575411782030338,0.6541695309608164,0.5834616840778439,0.4856873230575653
38
+ Swin,a,0.7181912302146427,0.6636277900031436,0.8782845303867403,0.22351233671988388
39
+ Swin,b,0.47780195057523284,0.8000628733102798,0.9174990791896869,0.326271186440678
40
+ Swin,c,1.1387689553203542,0.5369380697893744,0.8261528545119704,0.1729365524985963
41
+ Swin,d,0.0309918804128743,0.9871109713926438,0.9966482504604052,0.8825214899713467
42
+ Swin,e,0.5848406514011806,0.7464324917672887,0.8943237720426852,0.5714285714285714
43
+ Swin,f,0.6053860638443279,0.7410735032143134,0.9040542417311522,0.0843604491920022
44
+ Swin,g,0.24402792798448353,0.8985,0.999229,0.9078529278256923
45
+ Swin,h,0.5944505902426317,0.759,0.9974725555555555,0.8058017727639001
46
+ Swin,i,0.007144115434028208,0.9976666666666667,0.9999902222222222,0.9976720984369803
47
+ Swin,j,3.9496381425857545,0.419,0.11956033333333334,0.06591639871382636
48
+ Swin,k,3.7127543271519245,0.5181666666666667,0.4697328333333333,0.07841887153331208
49
+ Swin,l,1.5895203038408208,0.6710908994764951,0.5955901653865907,0.5130734304055112
50
+ CaiT,a,0.11189156073487824,0.967934611757309,0.9444355432780847,0.7243243243243244
51
+ CaiT,b,0.14077727915177515,0.9566174159069475,0.9445782688766113,0.6600985221674877
52
+ CaiT,c,0.1453998264081071,0.9556743162527507,0.9236316758747698,0.6552567237163814
53
+ CaiT,d,0.0895651844381368,0.9757937755422823,0.9521049723756906,0.7768115942028986
54
+ CaiT,e,0.5067908598680527,0.8507135016465422,0.8893589646560206,0.6633663366336634
55
+ CaiT,f,0.09735689565035,0.968553946247386,0.9382155086735556,0.39762611275964393
56
+ CaiT,g,0.04386970533267595,0.9845,0.9998996666666666,0.984726556084743
57
+ CaiT,h,0.04632043000892736,0.984,0.999908,0.9842416283650689
58
+ CaiT,i,0.016718765972414985,0.9946666666666667,0.9999668888888888,0.9946914399469144
59
+ CaiT,j,4.091754978463054,0.5111666666666667,0.502447,0.09726069559864574
60
+ CaiT,k,4.064604013383389,0.5213333333333333,0.4420845,0.09912170639899624
61
+ CaiT,l,1.3512259357719865,0.8281423510126381,0.7136882113330858,0.669379450661241
62
+ DeiT,a,0.2868140126428901,0.9104055328513047,0.9108996316758747,0.5060658578856152
63
+ DeiT,b,0.21093111757876476,0.9380697893744105,0.9419907918968692,0.5971370143149284
64
+ DeiT,c,0.4344255732631803,0.8610499842816725,0.8816003683241251,0.3978201634877384
65
+ DeiT,d,0.10285686065156817,0.977051241747878,0.9568195211786372,0.8
66
+ DeiT,e,0.6603749339457532,0.8210757409440176,0.8680541890562324,0.6417582417582418
67
+ DeiT,f,0.23529704651809683,0.9209975989466347,0.9196866062244752,0.2225609756097561
68
+ DeiT,g,0.07966256512608379,0.9715,0.9994743333333334,0.9722086786933203
69
+ DeiT,h,0.1981518727680668,0.9306666666666666,0.9987041666666666,0.9349796811503595
70
+ DeiT,i,0.022365192129276693,0.9921666666666666,0.9998618888888888,0.9922043456626306
71
+ DeiT,j,4.922376271247864,0.4945,0.5104821111111111,0.07839562443026436
72
+ DeiT,k,4.865078900694847,0.5151666666666667,0.5016916666666666,0.08146510893590149
73
+ DeiT,l,1.6993422816543697,0.7937708212151657,0.7000134906492581,0.6261503067484663
74
+ DeiT3,a,0.3992725303153578,0.9091480666457089,0.9363876611418048,0.5126475548060708
75
+ DeiT3,b,0.3124380833470795,0.9305249921408362,0.952292817679558,0.579047619047619
76
+ DeiT3,c,0.4766911079170793,0.8915435397673688,0.9285046040515654,0.46841294298921415
77
+ DeiT3,d,0.11976070332006253,0.9723357434768941,0.9784143646408839,0.7755102040816326
78
+ DeiT3,e,0.9584465352031193,0.7969264544456641,0.8915991826231742,0.621676891615542
79
+ DeiT3,f,0.33781881941231834,0.921617225621563,0.9456139627538378,0.23100303951367782
80
+ DeiT3,g,0.1398290316515031,0.9671666666666666,0.9996258333333332,0.9681590431550025
81
+ DeiT3,h,0.22691049999523785,0.9465,0.9994123333333333,0.9491364284582475
82
+ DeiT3,i,0.03767789686251308,0.9893333333333333,0.9998997777777778,0.9894284770399736
83
+ DeiT3,j,6.353702080726624,0.486,0.2588749444444445,0.06545454545454546
84
+ DeiT3,k,6.251550879061222,0.5081666666666667,0.38331972222222227,0.06820334701610357
85
+ DeiT3,l,2.203556089656545,0.7932949077256624,0.6245051702293716,0.6248200403109704
86
+ Twins_SVT,a,0.42451784632109274,0.8126375353662371,0.8918922651933701,0.3377777777777778
87
+ Twins_SVT,b,0.3914457758111573,0.8142093681232316,0.8958011049723758,0.3396648044692737
88
+ Twins_SVT,c,0.44262768309920586,0.8047783715812638,0.8853812154696132,0.3286486486486486
89
+ Twins_SVT,d,0.07794447650992994,0.9820811065702609,0.9897808471454881,0.8421052631578947
90
+ Twins_SVT,e,0.5991019170841715,0.712403951701427,0.818761825474911,0.5371024734982333
91
+ Twins_SVT,f,0.3470329082674194,0.8442413445898846,0.9101541579685174,0.13131749460043196
92
+ Twins_SVT,g,0.2309408655166626,0.9008333333333334,0.9874507777777777,0.9088681268188084
93
+ Twins_SVT,h,0.2580758459568024,0.8958333333333334,0.9890552222222222,0.9047110840067083
94
+ Twins_SVT,i,0.06473293882608414,0.9898333333333333,0.9991408888888889,0.9898248540450375
95
+ Twins_SVT,j,2.941682351350784,0.41483333333333333,0.16385111111111111,0.02823138665928591
96
+ Twins_SVT,k,2.775474429190159,0.5038333333333334,0.41604216666666666,0.03312763884378045
97
+ Twins_SVT,l,1.1202095961319853,0.7359737718788008,0.5944821592359222,0.5594282184770141
98
+ Twins_PCPVT,a,0.4347801183106501,0.7944042753850991,0.8711482504604051,0.2952586206896552
99
+ Twins_PCPVT,b,0.339161620420017,0.8575919522162841,0.9065580110497238,0.3768913342503439
100
+ Twins_PCPVT,c,0.467156688444122,0.7742848160955674,0.8578029465930019,0.2762096774193548
101
+ Twins_PCPVT,d,0.23605295665344328,0.9245520276642565,0.9460681399631676,0.5330739299610895
102
+ Twins_PCPVT,e,0.4829401325446714,0.7694840834248079,0.8557481268447741,0.5661157024793388
103
+ Twins_PCPVT,f,0.3720619066015847,0.8374254511656727,0.8931208308558979,0.11546565528866413
104
+ Twins_PCPVT,g,0.2366939251422882,0.9141666666666667,0.9769166666666668,0.9182928764080597
105
+ Twins_PCPVT,h,0.3045526731014252,0.87,0.9672449999999999,0.8812423873325214
106
+ Twins_PCPVT,i,0.18202911043167114,0.9496666666666667,0.9892225555555555,0.9504105090311987
107
+ Twins_PCPVT,j,1.547880547761917,0.4866666666666667,0.43741700000000006,0.1760299625468165
108
+ Twins_PCPVT,k,1.4932157402038575,0.5221666666666667,0.47360899999999995,0.18666666666666668
109
+ Twins_PCPVT,l,0.7149772635186269,0.7421606472423458,0.6893006569431472,0.579510175922732
110
+ PiT,a,0.44727815774318297,0.8010059729644766,0.9252882136279927,0.3329820864067439
111
+ PiT,b,0.3666396583074745,0.8374724929267526,0.9387182320441989,0.3793517406962785
112
+ PiT,c,0.587441599189167,0.7419050613014775,0.9032854511970534,0.27792436235708
113
+ PiT,d,0.033625746908820454,0.9855391386356491,0.9966703499079189,0.8729281767955801
114
+ PiT,e,0.631099864955006,0.7387486278814489,0.8873495799591311,0.5703971119133574
115
+ PiT,f,0.3817344946660815,0.8324684377662458,0.9379145273921177,0.127470754336426
116
+ PiT,g,0.19083814918994904,0.9163333333333333,0.9971276666666666,0.9226025285229725
117
+ PiT,h,0.30789998161792753,0.8656666666666667,0.9943662222222222,0.8812960235640648
118
+ PiT,i,0.01428527906537056,0.9948333333333333,0.9999181111111111,0.9948462177888612
119
+ PiT,j,5.1374336256980895,0.427,0.18367044444444444,0.031549295774647886
120
+ PiT,k,4.960880736589432,0.5055,0.6351591666666667,0.03637544657356284
121
+ PiT,l,1.8333862431563666,0.7295225001321982,0.6306325748279862,0.5562592174893728
122
+ Ensemble,a,,0.9239232945614586,0.9640395948434622,0.56
123
+ Ensemble,b,,0.926752593524049,0.9696408839779006,0.5693160813308688
124
+ Ensemble,c,,0.8563344860106885,0.9442707182320442,0.40261437908496733
125
+ Ensemble,d,,0.9874253379440427,0.9976040515653776,0.8850574712643678
126
+ Ensemble,e,,0.8254665203073546,0.9312873685007189,0.6595289079229122
127
+ Ensemble,f,,0.9207652389435366,0.9667325628328263,0.23140495867768596
128
+ Ensemble,g,,0.9656666666666667,0.999739111111111,0.9668063164679342
129
+ Ensemble,h,,0.9283333333333333,0.9995118888888889,0.9331259720062208
130
+ Ensemble,i,,0.9978333333333333,0.9999951111111111,0.9978380176284717
131
+ Ensemble,j,,0.47883333333333333,0.20988066666666666,0.04809741248097413
132
+ Ensemble,k,,0.511,0.48139255555555555,0.05109961190168176
133
+ Ensemble,l,,0.79144413304426,0.6240045673759117,0.6211335254562921
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18a90a3f0286609188cc447690fc72d8497817d89d062af359cb4c85b9eee87b
3
+ size 290985688
pit-gravit-a2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22f6a746eda79e9795369d2e0246f33a830c7797ec42d2eb547107648d6f5512
3
+ size 291039882
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22f6a746eda79e9795369d2e0246f33a830c7797ec42d2eb547107648d6f5512
3
+ size 291039882
roc_confusion_matrix/PiT_roc_confusion_matrix_a.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_b.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_c.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_d.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_e.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_f.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_g.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_h.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_i.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_j.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_k.png ADDED
roc_confusion_matrix/PiT_roc_confusion_matrix_l.png ADDED
roc_curves/PiT_ROC_a.png ADDED
roc_curves/PiT_ROC_b.png ADDED
roc_curves/PiT_ROC_c.png ADDED
roc_curves/PiT_ROC_d.png ADDED
roc_curves/PiT_ROC_e.png ADDED
roc_curves/PiT_ROC_f.png ADDED
roc_curves/PiT_ROC_g.png ADDED
roc_curves/PiT_ROC_h.png ADDED
roc_curves/PiT_ROC_i.png ADDED
roc_curves/PiT_ROC_j.png ADDED
roc_curves/PiT_ROC_k.png ADDED
roc_curves/PiT_ROC_l.png ADDED
training_curves/PiT_accuracy.png ADDED
training_curves/PiT_auc.png ADDED
training_curves/PiT_combined_metrics.png ADDED

Git LFS Details

  • SHA256: 6629748957dc99df82d4451a9f9a7bfea53035aabc0ef3ba21b4a7d1678f13ba
  • Pointer size: 131 Bytes
  • Size of remote file: 169 kB
training_curves/PiT_f1.png ADDED
training_curves/PiT_loss.png ADDED
training_curves/PiT_metrics.csv ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1
2
+ 1,0.3232941015481949,0.24128434872627258,0.8604375,0.905,0.9374816490625,0.967472,0.8621009077996665,0.900523560209424
3
+ 2,0.22616110746860504,0.173474432349205,0.9077,0.937,0.9695539621875,0.9835659999999999,0.9081820442675951,0.9376854599406528
4
+ 3,0.20072144094109534,0.17438986682891847,0.9180875,0.929,0.9758779828124999,0.984124,0.9183762440367201,0.9267285861713106
5
+ 4,0.1905623720407486,0.15850988340377808,0.921175,0.939,0.9781646125,0.985794,0.9214655773637541,0.9375639713408394
6
+ 5,0.1803186470091343,0.13645671474933624,0.92715,0.95,0.9804621574999999,0.9881040000000001,0.9275214525556523,0.9498997995991983
7
+ 6,0.17366552315354347,0.15974776780605315,0.9270625,0.936,0.9817743340625,0.98449,0.9273576097105509,0.9348268839103869
8
+ 7,0.16877106444835663,0.13136824417114257,0.930075,0.956,0.98282352875,0.989106,0.9302771961312195,0.9556451612903226
9
+ 8,0.16491497334241867,0.12256464439630509,0.9323625,0.96,0.9836165134374999,0.990048,0.9325185508511566,0.96
10
+ 9,0.15922530961632728,0.12970378270745278,0.934275,0.951,0.9847076121875,0.989506,0.9344732185139951,0.9513406156901688
11
+ 10,0.1583781011044979,0.15817884767055512,0.9346875,0.933,0.9848569975000001,0.990458,0.9348641809092836,0.9358851674641149
12
+ 11,0.15492646908760072,0.12343627136945724,0.9368625,0.959,0.9854855671874999,0.9904299999999999,0.937085061594609,0.9584599797365755
13
+ 12,0.15160933775901794,0.15969387048482894,0.9384375,0.937,0.9860790037499999,0.9904499999999999,0.9386116893315218,0.9398280802292264
14
+ 13,0.1486863681793213,0.11650157874822617,0.9393625,0.961,0.9866751465625001,0.9910080000000001,0.9394510528352285,0.9608826479438315
15
+ 14,0.14707621708512306,0.11402156364917755,0.9401875,0.96,0.9869002684375001,0.991412,0.9403611980107935,0.959758551307847
16
+ 15,0.14499566847085954,0.12285486936569213,0.9400125,0.959,0.9872670403125,0.9915059999999998,0.9401299948850381,0.9596059113300492
17
+ 16,0.1399480452030897,0.12467456012964248,0.9422,0.951,0.9881658053125,0.9921920000000001,0.9423081721771678,0.9519136408243376
18
+ 17,0.14094304995238782,0.13957817780971526,0.9422375,0.95,0.9879632543750001,0.991444,0.942395193158728,0.9517374517374517
19
+ 18,0.13952992499172687,0.10992657113075256,0.94285,0.969,0.9882162575,0.9917239999999999,0.9429270484845459,0.9689067201604814
20
+ 19,0.13795855878293514,0.11575935518741608,0.943825,0.964,0.988466728125,0.992878,0.9439413217573535,0.9644970414201184
21
+ 20,0.1346044053018093,0.10125552648305892,0.9454875,0.965,0.9890290790625,0.993032,0.9456105561167858,0.964824120603015
22
+ 21,0.13702923731207847,0.1645300726890564,0.9439,0.933,0.9886205840625,0.990704,0.9439882185557746,0.9357622243528284
23
+ 22,0.13661602101922035,0.12196814346313477,0.94505,0.957,0.9887078524999999,0.990478,0.9451569439592794,0.9572989076464746
24
+ 23,0.13644359414577484,0.12078015133738518,0.94375,0.96,0.9887120165625001,0.9915080000000001,0.9438776782819086,0.9603174603174603
25
+ 24,0.13201601221561432,0.12070372271537781,0.9464125,0.958,0.9894324925,0.991278,0.9465081167413248,0.958498023715415
26
+ 25,0.13056341710090638,0.1155894775390625,0.9476875,0.957,0.9896582925,0.991836,0.9477795385632822,0.9564336372847011
27
+ 26,0.12885473522245883,0.10226090061664582,0.9474,0.967,0.9899417896875,0.992904,0.9474105178964207,0.9668341708542714
28
+ 27,0.1208189173668623,0.10230878192186356,0.95155,0.965,0.9911893465625,0.99347,0.9516165272749969,0.9652432969215492
29
+ 28,0.12027741346657277,0.09924011027812958,0.9511625,0.966,0.9912573418749999,0.993736,0.951252074313449,0.9662027833001988
30
+ 29,0.1159165238648653,0.09786771166324615,0.953125,0.966,0.9919083196875002,0.993636,0.9531858583841006,0.9662027833001988
31
+ 30,0.11692634549736977,0.09434205585718156,0.95335,0.964,0.99174852625,0.994034,0.9534233582108179,0.9641434262948207
32
+ 31,0.11742365294694901,0.09521378380060196,0.9522875,0.969,0.9916503153125,0.9939039999999999,0.9523702566790202,0.9688442211055276
33
+ 32,0.11490899155139923,0.09616225984692574,0.9537875,0.969,0.9920278965625,0.9938380000000001,0.9538515310007365,0.9690309690309691
34
+ 33,0.11603512637019157,0.0948656535744667,0.9530875,0.968,0.9918384396875,0.99408,0.9531290979255911,0.9682539682539683
35
+ 34,0.11493948283791543,0.09457683789730072,0.9534125,0.969,0.992007243125,0.993876,0.9534700807750409,0.968968968968969
36
+ 35,0.11349499747157096,0.09559179240465164,0.954425,0.966,0.9922124087499999,0.993876,0.9544500524711409,0.9661354581673307
37
+ 36,0.11686774403452874,0.09406438562273979,0.95195,0.97,0.9917140234374999,0.994052,0.9520483009829849,0.9700598802395209
38
+ 37,0.11331877183020114,0.09683345186710357,0.9547125,0.968,0.9922413059375002,0.9938060000000001,0.9547470054083761,0.9681908548707754
39
+ 38,0.11312784470319748,0.09323691272735596,0.9550375,0.967,0.9922559184374999,0.993988,0.9550740023730718,0.9669669669669669
40
+ 39,0.11433582416772843,0.09317393565177917,0.9543,0.969,0.9920832596875001,0.993928,0.9543718643137059,0.9690309690309691
41
+ 40,0.1122991234511137,0.09810869953036308,0.9549125,0.969,0.9923716146875,0.994132,0.9549704755127773,0.9692765113974232
42
+ 41,0.11472533380389213,0.09223028540611267,0.954025,0.971,0.992018935625,0.994156,0.9540881288228685,0.971028971028971
43
+ 42,0.1101333582431078,0.09381417632102966,0.9553125,0.969,0.9926963296875,0.993932,0.9553710754634542,0.968968968968969
44
+ 43,0.1118443134278059,0.09355811506509781,0.9554625,0.969,0.9924281974999998,0.9944,0.9555042147986262,0.9692154915590864
45
+ 44,0.11458537501990795,0.09301779806613922,0.9532875,0.967,0.9920448515625001,0.99394,0.9533324175481099,0.967032967032967
46
+ 45,0.11242122393548488,0.0933422226011753,0.9547625,0.969,0.9923614246875,0.994114,0.9547958380694238,0.9690927218344965
47
+ 46,0.11411838567852974,0.09388426387310028,0.9534875,0.969,0.9920943803125,0.99377,0.9535531062374396,0.9689067201604814
48
+ 47,0.11184343811869621,0.09843983408808708,0.9544375,0.968,0.9924278096874999,0.994062,0.954483585369813,0.9683794466403162
49
+ 48,0.11007779505550862,0.09361994338035584,0.9558875,0.971,0.9926710259374999,0.994196,0.9559695067935969,0.971201588877855
50
+ 49,0.11184648263454437,0.09364327847957611,0.9543875,0.97,0.992441643125,0.99411,0.9544347739220559,0.9701195219123506
51
+ 50,0.11114003728330135,0.09281201779842377,0.9551625,0.969,0.9925268084375,0.9941559999999999,0.955195543286826,0.9690927218344965
52
+ 51,0.1113744495332241,0.09343778803944587,0.9555875,0.97,0.9924899793749999,0.99416,0.9556612132330001,0.9701195219123506
53
+ 52,0.11124969183802605,0.09389141094684601,0.955075,0.969,0.99251426875,0.994166,0.9551870324189526,0.9691542288557214
54
+ 53,0.1109674940675497,0.09317841828614473,0.955625,0.97,0.99255554875,0.99425,0.9556892505866493,0.9701195219123506
55
+ 54,0.10898926662504672,0.09370111536979675,0.956075,0.969,0.992826261875,0.9942519999999999,0.9561440730848912,0.9691542288557214
56
+ 55,0.11093568170666694,0.09370115166902541,0.954825,0.969,0.9925539775,0.9942460000000001,0.9549016671658181,0.9691542288557214
57
+ 56,0.10977310891151429,0.09387159994244576,0.9555,0.968,0.9927180124999999,0.9942099999999999,0.9555644315742173,0.9681908548707754
58
+ 57,0.11002452866435052,0.09189656394720078,0.95575,0.97,0.992683730625,0.994282,0.9558140695990813,0.9700598802395209
59
+ 58,0.10916990705430507,0.09243394762277603,0.956425,0.97,0.992814729375,0.9942259999999999,0.9564772273272073,0.97
60
+ 59,0.11117708411216735,0.09199372020363808,0.9547,0.97,0.992524945,0.994226,0.9547554246048591,0.97
61
+ 60,0.10969826458394527,0.09308649624884129,0.9555,0.969,0.992732808125,0.994266,0.9555488962141644,0.9691542288557214
62
+ 61,0.11091992988586426,0.09320096266269684,0.9554375,0.971,0.992555246875,0.9942300000000001,0.9555192334086117,0.9710867397806581
63
+ 62,0.11073220573961735,0.09430668842792511,0.9553125,0.968,0.9925707575,0.99419,0.9553443171740135,0.9681908548707754
64
+ 63,0.10991689996123313,0.09354863844811917,0.9560875,0.969,0.99268688125,0.994162,0.9561483441724608,0.9691542288557214
65
+ 64,0.11145925801694394,0.09202123552560806,0.9543875,0.969,0.9924811153125,0.994206,0.9544779750246385,0.968968968968969
66
+ 65,0.10793004957437516,0.09182925164699554,0.956525,0.97,0.992948348125,0.994226,0.9565792759051186,0.97
67
+ 66,0.10861873697042465,0.09277367499470711,0.955675,0.971,0.9928919128125,0.994214,0.9557192807192807,0.9710867397806581
68
+ 67,0.10996965856552124,0.09245775479078293,0.9556875,0.972,0.9926849209374999,0.9943259999999999,0.9557709822709636,0.9720558882235529
69
+ 68,0.11184505909979343,0.0927946150302887,0.9546375,0.97,0.99241495,0.9943160000000001,0.9546799875117078,0.9701195219123506
70
+ 69,0.11102755977511405,0.092463918030262,0.9548625,0.972,0.9925296890624999,0.9942859999999999,0.9548968911205206,0.9720558882235529
71
+ 70,0.10971400094330311,0.09227793508768081,0.9550125,0.971,0.9926966809375,0.9942240000000001,0.9550894093864257,0.9710867397806581
72
+ 71,0.11297933066189289,0.09311837112903595,0.9537125,0.968,0.992266400625,0.994306,0.9537754809072638,0.9681908548707754
73
+ 72,0.10945839685201644,0.09223374104499817,0.9563625,0.97,0.9927720190625,0.99427,0.956439275776444,0.9701195219123506
74
+ 73,0.10982317025661469,0.09159829139709473,0.9547875,0.97,0.9927048234374999,0.9943000000000001,0.9548163046058138,0.97
75
+ 74,0.10969363201260567,0.09145302675664425,0.9557,0.969,0.9927289915624999,0.994244,0.9557519914100931,0.968968968968969
76
+ 75,0.10969159477055072,0.09212042292952538,0.955,0.971,0.9927375403125,0.994282,0.9550842170929508,0.9710867397806581
77
+ 76,0.10900666551589966,0.09223489832878112,0.9560875,0.969,0.9928223765624999,0.994288,0.9561176691024921,0.9690927218344965
78
+ 77,0.10847178137600422,0.09184811854362487,0.9559125,0.969,0.9929062978125001,0.994298,0.9559669908488245,0.9690309690309691
79
+ 78,0.11163707442879676,0.09260461637377738,0.9549375,0.97,0.9924530112500001,0.994148,0.9549898243292173,0.9701195219123506
80
+ 79,0.1096200159728527,0.09222700503468513,0.9561,0.967,0.9927456696874999,0.99416,0.9561799715519177,0.967032967032967
81
+ 80,0.10934312551617623,0.09274832218885422,0.9566625,0.97,0.9927783287499999,0.994122,0.9567398275582396,0.9701195219123506
82
+ 81,0.10930199634432793,0.09384609511494636,0.9558,0.968,0.9927701359375,0.9941400000000001,0.95583091835715,0.9681908548707754
83
+ 82,0.11267508911788464,0.09524662458896636,0.95495,0.968,0.992286083125,0.99398,0.9549837621783662,0.9681908548707754
84
+ 83,0.1086170636922121,0.09409611165523529,0.9565375,0.968,0.9928849203125001,0.994068,0.9565478198927755,0.9681908548707754
85
+ 84,0.1119013113334775,0.09427453267574311,0.9552625,0.966,0.992415096875,0.9941059999999999,0.955328948189568,0.9662698412698413
86
+ 85,0.10950523447096348,0.09191533011198044,0.9561875,0.966,0.9927495696875,0.9942000000000001,0.9562340013735406,0.966
87
+ 86,0.11022727983891964,0.0927174622118473,0.955625,0.968,0.99265143,0.994222,0.9556803995006242,0.9681908548707754
88
+ 87,0.10824401076734066,0.09243075692653656,0.9562,0.969,0.9929358040625,0.994182,0.9562623261851861,0.9690927218344965
89
+ 88,0.10982458600103855,0.09261692684888839,0.955375,0.968,0.9926980818750001,0.9942,0.9554262598012286,0.9681274900398407
90
+ 89,0.1089077548712492,0.09330538702011108,0.9558875,0.968,0.992813924375,0.994302,0.9559255142439645,0.9681908548707754
91
+ 90,0.1095367282152176,0.09203421288728714,0.95615,0.969,0.9927390484374999,0.994312,0.9562014632806453,0.9691542288557214
92
+ 91,0.1090563885629177,0.09258661818504334,0.9561875,0.969,0.992808510625,0.994286,0.9562547583090998,0.9691542288557214
93
+ 92,0.10825774060487747,0.09169195887446403,0.95635,0.97,0.9929186615625,0.9942620000000001,0.9563609097725568,0.9701195219123506
94
+ 93,0.11120219169855118,0.09436736851930619,0.95515,0.968,0.9924979953124999,0.9942840000000001,0.9552350533985428,0.9682539682539683
95
+ 94,0.10998510930538177,0.09147375085949898,0.9552625,0.971,0.9926854624999999,0.9942939999999999,0.9553267178431005,0.971028971028971
training_metrics.csv ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1
2
+ 1,0.3232941015481949,0.24128434872627258,0.8604375,0.905,0.9374816490625,0.967472,0.8621009077996665,0.900523560209424
3
+ 2,0.22616110746860504,0.173474432349205,0.9077,0.937,0.9695539621875,0.9835659999999999,0.9081820442675951,0.9376854599406528
4
+ 3,0.20072144094109534,0.17438986682891847,0.9180875,0.929,0.9758779828124999,0.984124,0.9183762440367201,0.9267285861713106
5
+ 4,0.1905623720407486,0.15850988340377808,0.921175,0.939,0.9781646125,0.985794,0.9214655773637541,0.9375639713408394
6
+ 5,0.1803186470091343,0.13645671474933624,0.92715,0.95,0.9804621574999999,0.9881040000000001,0.9275214525556523,0.9498997995991983
7
+ 6,0.17366552315354347,0.15974776780605315,0.9270625,0.936,0.9817743340625,0.98449,0.9273576097105509,0.9348268839103869
8
+ 7,0.16877106444835663,0.13136824417114257,0.930075,0.956,0.98282352875,0.989106,0.9302771961312195,0.9556451612903226
9
+ 8,0.16491497334241867,0.12256464439630509,0.9323625,0.96,0.9836165134374999,0.990048,0.9325185508511566,0.96
10
+ 9,0.15922530961632728,0.12970378270745278,0.934275,0.951,0.9847076121875,0.989506,0.9344732185139951,0.9513406156901688
11
+ 10,0.1583781011044979,0.15817884767055512,0.9346875,0.933,0.9848569975000001,0.990458,0.9348641809092836,0.9358851674641149
12
+ 11,0.15492646908760072,0.12343627136945724,0.9368625,0.959,0.9854855671874999,0.9904299999999999,0.937085061594609,0.9584599797365755
13
+ 12,0.15160933775901794,0.15969387048482894,0.9384375,0.937,0.9860790037499999,0.9904499999999999,0.9386116893315218,0.9398280802292264
14
+ 13,0.1486863681793213,0.11650157874822617,0.9393625,0.961,0.9866751465625001,0.9910080000000001,0.9394510528352285,0.9608826479438315
15
+ 14,0.14707621708512306,0.11402156364917755,0.9401875,0.96,0.9869002684375001,0.991412,0.9403611980107935,0.959758551307847
16
+ 15,0.14499566847085954,0.12285486936569213,0.9400125,0.959,0.9872670403125,0.9915059999999998,0.9401299948850381,0.9596059113300492
17
+ 16,0.1399480452030897,0.12467456012964248,0.9422,0.951,0.9881658053125,0.9921920000000001,0.9423081721771678,0.9519136408243376
18
+ 17,0.14094304995238782,0.13957817780971526,0.9422375,0.95,0.9879632543750001,0.991444,0.942395193158728,0.9517374517374517
19
+ 18,0.13952992499172687,0.10992657113075256,0.94285,0.969,0.9882162575,0.9917239999999999,0.9429270484845459,0.9689067201604814
20
+ 19,0.13795855878293514,0.11575935518741608,0.943825,0.964,0.988466728125,0.992878,0.9439413217573535,0.9644970414201184
21
+ 20,0.1346044053018093,0.10125552648305892,0.9454875,0.965,0.9890290790625,0.993032,0.9456105561167858,0.964824120603015
22
+ 21,0.13702923731207847,0.1645300726890564,0.9439,0.933,0.9886205840625,0.990704,0.9439882185557746,0.9357622243528284
23
+ 22,0.13661602101922035,0.12196814346313477,0.94505,0.957,0.9887078524999999,0.990478,0.9451569439592794,0.9572989076464746
24
+ 23,0.13644359414577484,0.12078015133738518,0.94375,0.96,0.9887120165625001,0.9915080000000001,0.9438776782819086,0.9603174603174603
25
+ 24,0.13201601221561432,0.12070372271537781,0.9464125,0.958,0.9894324925,0.991278,0.9465081167413248,0.958498023715415
26
+ 25,0.13056341710090638,0.1155894775390625,0.9476875,0.957,0.9896582925,0.991836,0.9477795385632822,0.9564336372847011
27
+ 26,0.12885473522245883,0.10226090061664582,0.9474,0.967,0.9899417896875,0.992904,0.9474105178964207,0.9668341708542714
28
+ 27,0.1208189173668623,0.10230878192186356,0.95155,0.965,0.9911893465625,0.99347,0.9516165272749969,0.9652432969215492
29
+ 28,0.12027741346657277,0.09924011027812958,0.9511625,0.966,0.9912573418749999,0.993736,0.951252074313449,0.9662027833001988
30
+ 29,0.1159165238648653,0.09786771166324615,0.953125,0.966,0.9919083196875002,0.993636,0.9531858583841006,0.9662027833001988
31
+ 30,0.11692634549736977,0.09434205585718156,0.95335,0.964,0.99174852625,0.994034,0.9534233582108179,0.9641434262948207
32
+ 31,0.11742365294694901,0.09521378380060196,0.9522875,0.969,0.9916503153125,0.9939039999999999,0.9523702566790202,0.9688442211055276
33
+ 32,0.11490899155139923,0.09616225984692574,0.9537875,0.969,0.9920278965625,0.9938380000000001,0.9538515310007365,0.9690309690309691
34
+ 33,0.11603512637019157,0.0948656535744667,0.9530875,0.968,0.9918384396875,0.99408,0.9531290979255911,0.9682539682539683
35
+ 34,0.11493948283791543,0.09457683789730072,0.9534125,0.969,0.992007243125,0.993876,0.9534700807750409,0.968968968968969
36
+ 35,0.11349499747157096,0.09559179240465164,0.954425,0.966,0.9922124087499999,0.993876,0.9544500524711409,0.9661354581673307
37
+ 36,0.11686774403452874,0.09406438562273979,0.95195,0.97,0.9917140234374999,0.994052,0.9520483009829849,0.9700598802395209
38
+ 37,0.11331877183020114,0.09683345186710357,0.9547125,0.968,0.9922413059375002,0.9938060000000001,0.9547470054083761,0.9681908548707754
39
+ 38,0.11312784470319748,0.09323691272735596,0.9550375,0.967,0.9922559184374999,0.993988,0.9550740023730718,0.9669669669669669
40
+ 39,0.11433582416772843,0.09317393565177917,0.9543,0.969,0.9920832596875001,0.993928,0.9543718643137059,0.9690309690309691
41
+ 40,0.1122991234511137,0.09810869953036308,0.9549125,0.969,0.9923716146875,0.994132,0.9549704755127773,0.9692765113974232
42
+ 41,0.11472533380389213,0.09223028540611267,0.954025,0.971,0.992018935625,0.994156,0.9540881288228685,0.971028971028971
43
+ 42,0.1101333582431078,0.09381417632102966,0.9553125,0.969,0.9926963296875,0.993932,0.9553710754634542,0.968968968968969
44
+ 43,0.1118443134278059,0.09355811506509781,0.9554625,0.969,0.9924281974999998,0.9944,0.9555042147986262,0.9692154915590864
45
+ 44,0.11458537501990795,0.09301779806613922,0.9532875,0.967,0.9920448515625001,0.99394,0.9533324175481099,0.967032967032967
46
+ 45,0.11242122393548488,0.0933422226011753,0.9547625,0.969,0.9923614246875,0.994114,0.9547958380694238,0.9690927218344965
47
+ 46,0.11411838567852974,0.09388426387310028,0.9534875,0.969,0.9920943803125,0.99377,0.9535531062374396,0.9689067201604814
48
+ 47,0.11184343811869621,0.09843983408808708,0.9544375,0.968,0.9924278096874999,0.994062,0.954483585369813,0.9683794466403162
49
+ 48,0.11007779505550862,0.09361994338035584,0.9558875,0.971,0.9926710259374999,0.994196,0.9559695067935969,0.971201588877855
50
+ 49,0.11184648263454437,0.09364327847957611,0.9543875,0.97,0.992441643125,0.99411,0.9544347739220559,0.9701195219123506
51
+ 50,0.11114003728330135,0.09281201779842377,0.9551625,0.969,0.9925268084375,0.9941559999999999,0.955195543286826,0.9690927218344965
52
+ 51,0.1113744495332241,0.09343778803944587,0.9555875,0.97,0.9924899793749999,0.99416,0.9556612132330001,0.9701195219123506
53
+ 52,0.11124969183802605,0.09389141094684601,0.955075,0.969,0.99251426875,0.994166,0.9551870324189526,0.9691542288557214
54
+ 53,0.1109674940675497,0.09317841828614473,0.955625,0.97,0.99255554875,0.99425,0.9556892505866493,0.9701195219123506
55
+ 54,0.10898926662504672,0.09370111536979675,0.956075,0.969,0.992826261875,0.9942519999999999,0.9561440730848912,0.9691542288557214
56
+ 55,0.11093568170666694,0.09370115166902541,0.954825,0.969,0.9925539775,0.9942460000000001,0.9549016671658181,0.9691542288557214
57
+ 56,0.10977310891151429,0.09387159994244576,0.9555,0.968,0.9927180124999999,0.9942099999999999,0.9555644315742173,0.9681908548707754
58
+ 57,0.11002452866435052,0.09189656394720078,0.95575,0.97,0.992683730625,0.994282,0.9558140695990813,0.9700598802395209
59
+ 58,0.10916990705430507,0.09243394762277603,0.956425,0.97,0.992814729375,0.9942259999999999,0.9564772273272073,0.97
60
+ 59,0.11117708411216735,0.09199372020363808,0.9547,0.97,0.992524945,0.994226,0.9547554246048591,0.97
61
+ 60,0.10969826458394527,0.09308649624884129,0.9555,0.969,0.992732808125,0.994266,0.9555488962141644,0.9691542288557214
62
+ 61,0.11091992988586426,0.09320096266269684,0.9554375,0.971,0.992555246875,0.9942300000000001,0.9555192334086117,0.9710867397806581
63
+ 62,0.11073220573961735,0.09430668842792511,0.9553125,0.968,0.9925707575,0.99419,0.9553443171740135,0.9681908548707754
64
+ 63,0.10991689996123313,0.09354863844811917,0.9560875,0.969,0.99268688125,0.994162,0.9561483441724608,0.9691542288557214
65
+ 64,0.11145925801694394,0.09202123552560806,0.9543875,0.969,0.9924811153125,0.994206,0.9544779750246385,0.968968968968969
66
+ 65,0.10793004957437516,0.09182925164699554,0.956525,0.97,0.992948348125,0.994226,0.9565792759051186,0.97
67
+ 66,0.10861873697042465,0.09277367499470711,0.955675,0.971,0.9928919128125,0.994214,0.9557192807192807,0.9710867397806581
68
+ 67,0.10996965856552124,0.09245775479078293,0.9556875,0.972,0.9926849209374999,0.9943259999999999,0.9557709822709636,0.9720558882235529
69
+ 68,0.11184505909979343,0.0927946150302887,0.9546375,0.97,0.99241495,0.9943160000000001,0.9546799875117078,0.9701195219123506
70
+ 69,0.11102755977511405,0.092463918030262,0.9548625,0.972,0.9925296890624999,0.9942859999999999,0.9548968911205206,0.9720558882235529
71
+ 70,0.10971400094330311,0.09227793508768081,0.9550125,0.971,0.9926966809375,0.9942240000000001,0.9550894093864257,0.9710867397806581
72
+ 71,0.11297933066189289,0.09311837112903595,0.9537125,0.968,0.992266400625,0.994306,0.9537754809072638,0.9681908548707754
73
+ 72,0.10945839685201644,0.09223374104499817,0.9563625,0.97,0.9927720190625,0.99427,0.956439275776444,0.9701195219123506
74
+ 73,0.10982317025661469,0.09159829139709473,0.9547875,0.97,0.9927048234374999,0.9943000000000001,0.9548163046058138,0.97
75
+ 74,0.10969363201260567,0.09145302675664425,0.9557,0.969,0.9927289915624999,0.994244,0.9557519914100931,0.968968968968969
76
+ 75,0.10969159477055072,0.09212042292952538,0.955,0.971,0.9927375403125,0.994282,0.9550842170929508,0.9710867397806581
77
+ 76,0.10900666551589966,0.09223489832878112,0.9560875,0.969,0.9928223765624999,0.994288,0.9561176691024921,0.9690927218344965
78
+ 77,0.10847178137600422,0.09184811854362487,0.9559125,0.969,0.9929062978125001,0.994298,0.9559669908488245,0.9690309690309691
79
+ 78,0.11163707442879676,0.09260461637377738,0.9549375,0.97,0.9924530112500001,0.994148,0.9549898243292173,0.9701195219123506
80
+ 79,0.1096200159728527,0.09222700503468513,0.9561,0.967,0.9927456696874999,0.99416,0.9561799715519177,0.967032967032967
81
+ 80,0.10934312551617623,0.09274832218885422,0.9566625,0.97,0.9927783287499999,0.994122,0.9567398275582396,0.9701195219123506
82
+ 81,0.10930199634432793,0.09384609511494636,0.9558,0.968,0.9927701359375,0.9941400000000001,0.95583091835715,0.9681908548707754
83
+ 82,0.11267508911788464,0.09524662458896636,0.95495,0.968,0.992286083125,0.99398,0.9549837621783662,0.9681908548707754
84
+ 83,0.1086170636922121,0.09409611165523529,0.9565375,0.968,0.9928849203125001,0.994068,0.9565478198927755,0.9681908548707754
85
+ 84,0.1119013113334775,0.09427453267574311,0.9552625,0.966,0.992415096875,0.9941059999999999,0.955328948189568,0.9662698412698413
86
+ 85,0.10950523447096348,0.09191533011198044,0.9561875,0.966,0.9927495696875,0.9942000000000001,0.9562340013735406,0.966
87
+ 86,0.11022727983891964,0.0927174622118473,0.955625,0.968,0.99265143,0.994222,0.9556803995006242,0.9681908548707754
88
+ 87,0.10824401076734066,0.09243075692653656,0.9562,0.969,0.9929358040625,0.994182,0.9562623261851861,0.9690927218344965
89
+ 88,0.10982458600103855,0.09261692684888839,0.955375,0.968,0.9926980818750001,0.9942,0.9554262598012286,0.9681274900398407
90
+ 89,0.1089077548712492,0.09330538702011108,0.9558875,0.968,0.992813924375,0.994302,0.9559255142439645,0.9681908548707754
91
+ 90,0.1095367282152176,0.09203421288728714,0.95615,0.969,0.9927390484374999,0.994312,0.9562014632806453,0.9691542288557214
92
+ 91,0.1090563885629177,0.09258661818504334,0.9561875,0.969,0.992808510625,0.994286,0.9562547583090998,0.9691542288557214
93
+ 92,0.10825774060487747,0.09169195887446403,0.95635,0.97,0.9929186615625,0.9942620000000001,0.9563609097725568,0.9701195219123506
94
+ 93,0.11120219169855118,0.09436736851930619,0.95515,0.968,0.9924979953124999,0.9942840000000001,0.9552350533985428,0.9682539682539683
95
+ 94,0.10998510930538177,0.09147375085949898,0.9552625,0.971,0.9926854624999999,0.9942939999999999,0.9553267178431005,0.971028971028971