parlange commited on
Commit
8dfbd8b
·
verified ·
1 Parent(s): 591bb2b

Upload PiT model from experiment c2

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 +166 -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-c2.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 +79 -0
  50. training_metrics.csv +79 -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_c2.ipynb filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,166 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - image-classification
5
+ - pytorch
6
+ - timm
7
+ - pit
8
+ - vision-transformer
9
+ - transformer
10
+ - gravitational-lensing
11
+ - strong-lensing
12
+ - astronomy
13
+ - astrophysics
14
+ datasets:
15
+ - parlange/gravit-c21-j24
16
+ metrics:
17
+ - accuracy
18
+ - auc
19
+ - f1
20
+ paper:
21
+ - title: "GraViT: A Gravitational Lens Discovery Toolkit with Vision Transformers"
22
+ url: "https://arxiv.org/abs/2509.00226"
23
+ authors: "Parlange et al."
24
+ model-index:
25
+ - name: PiT-c2
26
+ results:
27
+ - task:
28
+ type: image-classification
29
+ name: Strong Gravitational Lens Discovery
30
+ dataset:
31
+ type: common-test-sample
32
+ name: Common Test Sample (More et al. 2024)
33
+ metrics:
34
+ - type: accuracy
35
+ value: 0.8074
36
+ name: Average Accuracy
37
+ - type: auc
38
+ value: 0.8443
39
+ name: Average AUC-ROC
40
+ - type: f1
41
+ value: 0.5437
42
+ name: Average F1-Score
43
+ ---
44
+
45
+ # 🌌 pit-gravit-c2
46
+
47
+ 🔭 This model is part of **GraViT**: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery
48
+
49
+ 🔗 **GitHub Repository**: [https://github.com/parlange/gravit](https://github.com/parlange/gravit)
50
+
51
+ ## 🛰️ Model Details
52
+
53
+ - **🤖 Model Type**: PiT
54
+ - **🧪 Experiment**: C2 - C21+J24-half
55
+ - **🌌 Dataset**: C21+J24
56
+ - **🪐 Fine-tuning Strategy**: half
57
+
58
+
59
+
60
+ ## 💻 Quick Start
61
+
62
+ ```python
63
+ import torch
64
+ import timm
65
+
66
+ # Load the model directly from the Hub
67
+ model = timm.create_model(
68
+ 'hf-hub:parlange/pit-gravit-c2',
69
+ pretrained=True
70
+ )
71
+ model.eval()
72
+
73
+ # Example inference
74
+ dummy_input = torch.randn(1, 3, 224, 224)
75
+ with torch.no_grad():
76
+ output = model(dummy_input)
77
+ predictions = torch.softmax(output, dim=1)
78
+ print(f"Lens probability: {predictions[0][1]:.4f}")
79
+ ```
80
+
81
+ ## ⚡️ Training Configuration
82
+
83
+ **Training Dataset:** C21+J24 (Cañameras et al. 2021 + Jaelani et al. 2024)
84
+ **Fine-tuning Strategy:** half
85
+
86
+
87
+ | 🔧 Parameter | 📝 Value |
88
+ |--------------|----------|
89
+ | Batch Size | 192 |
90
+ | Learning Rate | AdamW with ReduceLROnPlateau |
91
+ | Epochs | 100 |
92
+ | Patience | 10 |
93
+ | Optimizer | AdamW |
94
+ | Scheduler | ReduceLROnPlateau |
95
+ | Image Size | 224x224 |
96
+ | Fine Tune Mode | half |
97
+ | Stochastic Depth Probability | 0.1 |
98
+
99
+
100
+ ## 📈 Training Curves
101
+
102
+ ![Combined Training Metrics](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/training_curves/PiT_combined_metrics.png)
103
+
104
+
105
+ ## 🏁 Final Epoch Training Metrics
106
+
107
+ | Metric | Training | Validation |
108
+ |:---------:|:-----------:|:-------------:|
109
+ | 📉 Loss | 0.1768 | 0.1241 |
110
+ | 🎯 Accuracy | 0.9266 | 0.9534 |
111
+ | 📊 AUC-ROC | 0.9810 | 0.9902 |
112
+ | ⚖️ F1 Score | 0.9263 | 0.9533 |
113
+
114
+
115
+ ## ☑️ Evaluation Results
116
+
117
+ ### ROC Curves and Confusion Matrices
118
+
119
+ Performance across all test datasets (a through l) in the Common Test Sample (More et al. 2024):
120
+
121
+ ![ROC + Confusion Matrix - Dataset A](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_a.png)
122
+ ![ROC + Confusion Matrix - Dataset B](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_b.png)
123
+ ![ROC + Confusion Matrix - Dataset C](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_c.png)
124
+ ![ROC + Confusion Matrix - Dataset D](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_d.png)
125
+ ![ROC + Confusion Matrix - Dataset E](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_e.png)
126
+ ![ROC + Confusion Matrix - Dataset F](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_f.png)
127
+ ![ROC + Confusion Matrix - Dataset G](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_g.png)
128
+ ![ROC + Confusion Matrix - Dataset H](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_h.png)
129
+ ![ROC + Confusion Matrix - Dataset I](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_i.png)
130
+ ![ROC + Confusion Matrix - Dataset J](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_j.png)
131
+ ![ROC + Confusion Matrix - Dataset K](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_k.png)
132
+ ![ROC + Confusion Matrix - Dataset L](https://huggingface.co/parlange/pit-gravit-c2/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_l.png)
133
+
134
+ ### 📋 Performance Summary
135
+
136
+ Average performance across 12 test datasets from the Common Test Sample (More et al. 2024):
137
+
138
+ | Metric | Value |
139
+ |-----------|----------|
140
+ | 🎯 Average Accuracy | 0.8074 |
141
+ | 📈 Average AUC-ROC | 0.8443 |
142
+ | ⚖️ Average F1-Score | 0.5437 |
143
+
144
+
145
+ ## 📘 Citation
146
+
147
+ If you use this model in your research, please cite:
148
+
149
+ ```bibtex
150
+ @misc{parlange2025gravit,
151
+ title={GraViT: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery},
152
+ 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},
153
+ year={2025},
154
+ eprint={2509.00226},
155
+ archivePrefix={arXiv},
156
+ primaryClass={cs.CV},
157
+ url={https://arxiv.org/abs/2509.00226},
158
+ }
159
+ ```
160
+
161
+ ---
162
+
163
+
164
+ ## Model Card Contact
165
+
166
+ 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_c2",
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_c2",
60
+ "experiment": "c2",
61
+ "training_strategy": "half",
62
+ "dataset": "C21+J24",
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-c2",
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.35447569389258865,0.8949115044247787,0.9020846228498507,0.7480106100795756
3
+ ViT,b,0.2228443425890036,0.9264382269726501,0.9263609576427256,0.5465116279069767
4
+ ViT,c,0.46229862214933587,0.8349575605155611,0.8684438305709024,0.34944237918215615
5
+ ViT,d,0.11673173789463115,0.9537881169443572,0.9704917127071824,0.6573426573426573
6
+ ViT,e,0.3652098159562351,0.8825466520307355,0.920767426019829,0.7249357326478149
7
+ ViT,f,0.24608832064126923,0.9108519842016175,0.9203345361214339,0.22926829268292684
8
+ ViT,g,0.10483654439449311,0.9635,0.997473,0.9644999189495866
9
+ ViT,h,0.23178722894191742,0.915,0.9939590555555555,0.9210526315789473
10
+ ViT,i,0.04857917896906535,0.978,0.9990572222222223,0.9782966129562644
11
+ ViT,j,2.494326035181681,0.6106666666666667,0.5831323333333334,0.42349457058242845
12
+ ViT,k,2.4380686638752618,0.6251666666666666,0.7805802777777779,0.43278688524590164
13
+ ViT,l,1.0272723743838732,0.8127329565949261,0.7993805230717175,0.7184308053873272
14
+ MLP-Mixer,a,1.230455079964832,0.6227876106194691,0.8958911227772556,0.49028400597907323
15
+ MLP-Mixer,b,1.0728926989350893,0.7004086765168186,0.9182900552486188,0.25604996096799376
16
+ MLP-Mixer,c,1.374837134027586,0.5576862621817039,0.8979152854511969,0.18904899135446687
17
+ MLP-Mixer,d,0.09552026474693218,0.9603898145237346,0.9868913443830571,0.7224669603524229
18
+ MLP-Mixer,e,0.9593323631422711,0.7069154774972558,0.9188677817301143,0.5512605042016807
19
+ MLP-Mixer,f,0.9257462782946794,0.7154410381794245,0.9306221006103087,0.09779367918902802
20
+ MLP-Mixer,g,0.5643243643840155,0.8425,0.991425611111111,0.8635773061931572
21
+ MLP-Mixer,h,0.7244052359660467,0.7668333333333334,0.9891666111111111,0.8104592873594364
22
+ MLP-Mixer,i,0.04615406060218811,0.9803333333333333,0.9994367777777778,0.980655737704918
23
+ MLP-Mixer,j,3.0292422666549683,0.45216666666666666,0.392282,0.28309705561613957
24
+ MLP-Mixer,k,2.5110719747940697,0.59,0.7661271111111111,0.3453964874933475
25
+ MLP-Mixer,l,1.4846716919555334,0.6762053625105207,0.7295511702036557,0.5855010004617516
26
+ CvT,a,0.7465745627352621,0.6493362831858407,0.7317079694031161,0.4389380530973451
27
+ CvT,b,0.7336456650122649,0.6765168186104998,0.7552670349907918,0.1942051683633516
28
+ CvT,c,0.8642418710588097,0.5919522162841874,0.6964806629834255,0.16041397153945666
29
+ CvT,d,0.06205783033066015,0.9761081420936812,0.9876427255985267,0.7654320987654321
30
+ CvT,e,0.6019917449757506,0.7178924259055982,0.7936123514720351,0.4910891089108911
31
+ CvT,f,0.5685286294680824,0.7414895617829603,0.8061353821076506,0.08274941608274941
32
+ CvT,g,0.4509977758725484,0.8055,0.9201512777777776,0.8277999114652501
33
+ CvT,h,0.5202355206807454,0.7606666666666667,0.9072719444444444,0.7961964235026966
34
+ CvT,i,0.09494428576032321,0.9643333333333334,0.9977035555555557,0.9632554945054945
35
+ CvT,j,2.988422914981842,0.3456666666666667,0.14668444444444442,0.022896963663514187
36
+ CvT,k,2.6323694267769655,0.5045,0.6181494444444444,0.0300163132137031
37
+ CvT,l,1.337245315202257,0.645425033064807,0.6032419706344807,0.5021944632005402
38
+ Swin,a,0.47572549887463056,0.8407079646017699,0.905882487792577,0.6742081447963801
39
+ Swin,b,0.24361524523634911,0.9163784973278843,0.9362615101289135,0.5283687943262412
40
+ Swin,c,0.4370936370240709,0.8535051870480981,0.9087605893186003,0.3900523560209424
41
+ Swin,d,0.038348094671021904,0.9880540710468406,0.9911620626151013,0.8869047619047619
42
+ Swin,e,0.3579506372581067,0.8781558726673985,0.9260273972602739,0.7286063569682152
43
+ Swin,f,0.24650774364781286,0.9156479217603912,0.9413092437445593,0.24937238493723848
44
+ Swin,g,0.11494702147444089,0.9593333333333334,0.9989898888888888,0.9607969151670951
45
+ Swin,h,0.2175228010714054,0.926,0.9979807777777777,0.9308841843088418
46
+ Swin,i,0.006121216081082821,0.9973333333333333,0.9999798888888889,0.9973315543695798
47
+ Swin,j,2.5422419211069744,0.5825,0.4893003333333333,0.3679031037093111
48
+ Swin,k,2.433416116627554,0.6205,0.7913794999999999,0.39036144578313253
49
+ Swin,l,1.035569912688268,0.8089455332451605,0.7797953948083542,0.7088143668682426
50
+ CaiT,a,0.3509529214517205,0.9081858407079646,0.8966973093999068,0.7726027397260274
51
+ CaiT,b,0.1907231829655279,0.9380697893744105,0.9234548802946593,0.5887265135699373
52
+ CaiT,c,0.3048490960337163,0.90883370009431,0.8791160220994475,0.493006993006993
53
+ CaiT,d,0.06549901952829443,0.9849104055328513,0.969243093922652,0.8545454545454545
54
+ CaiT,e,0.31167979835318943,0.9187705817782656,0.9264058124574283,0.7921348314606742
55
+ CaiT,f,0.1541684599891403,0.9499717886025955,0.9222261921687871,0.3464373464373464
56
+ CaiT,g,0.07805611325552066,0.9708333333333333,0.9986172777777778,0.9714937286202965
57
+ CaiT,h,0.13856186520308256,0.9553333333333334,0.997130611111111,0.9569961489088575
58
+ CaiT,i,0.011666435472667217,0.9956666666666667,0.9999013333333333,0.9956594323873121
59
+ CaiT,j,1.8389671653707822,0.6116666666666667,0.7423962222222222,0.4151606425702811
60
+ CaiT,k,1.7725774958133698,0.6365,0.8888650555555555,0.4312907431551499
61
+ CaiT,l,0.7395369254032035,0.8362991463268006,0.8693810723675515,0.7436693965922997
62
+ DeiT,a,0.48058320357736234,0.8263274336283186,0.8941450218931248,0.6594360086767896
63
+ DeiT,b,0.23002449519573911,0.9251807607670544,0.9313581952117864,0.5608856088560885
64
+ DeiT,c,0.49494195908204974,0.8154668343288274,0.8907605893186004,0.34118967452300786
65
+ DeiT,d,0.05036040664735698,0.9849104055328513,0.9769023941068141,0.8636363636363636
66
+ DeiT,e,0.338863200106291,0.8792535675082327,0.9161961704382048,0.7342995169082126
67
+ DeiT,f,0.26403015722496653,0.9037050968591311,0.9291450866890099,0.2289156626506024
68
+ DeiT,g,0.10851164469867945,0.9641666666666666,0.9990410000000001,0.9653393519264872
69
+ DeiT,h,0.2489620513096452,0.906,0.9981344444444444,0.9139194139194139
70
+ DeiT,i,0.013259729760388533,0.9958333333333333,0.9998315555555556,0.9958423415932147
71
+ DeiT,j,1.2026229511300723,0.7143333333333334,0.7246498888888889,0.6356292517006803
72
+ DeiT,k,1.1073710439900557,0.746,0.8698901111111111,0.6623836951705804
73
+ DeiT,l,0.5658274294531473,0.8476012985451485,0.867833726587774,0.7854785478547854
74
+ DeiT3,a,0.39277621998196155,0.8661504424778761,0.9195532732705195,0.7125890736342043
75
+ DeiT3,b,0.338128161960636,0.8824269097767997,0.9331012891344382,0.44510385756676557
76
+ DeiT3,c,0.323060417608134,0.8883998742533794,0.922292817679558,0.4580152671755725
77
+ DeiT3,d,0.12409640010358478,0.9553599497013517,0.9608121546961326,0.6787330316742082
78
+ DeiT3,e,0.24973662732461413,0.9209659714599341,0.9483084840687203,0.8064516129032258
79
+ DeiT3,f,0.2540075041596123,0.9116042881324055,0.9380772021883802,0.24193548387096775
80
+ DeiT3,g,0.1656125110021482,0.9416666666666667,0.9990236666666666,0.944760101010101
81
+ DeiT3,h,0.15762409150910875,0.9448333333333333,0.9990646111111111,0.9476017096723128
82
+ DeiT3,i,0.05214000094247361,0.9803333333333333,0.9997376666666667,0.9806684141546527
83
+ DeiT3,j,1.1591287109454473,0.696,0.7744774999999999,0.6248457424928013
84
+ DeiT3,k,1.0456561943689981,0.7346666666666667,0.845634,0.6561555075593952
85
+ DeiT3,l,0.5223108836063022,0.854033906456655,0.8898184372191467,0.7933968686181075
86
+ Twins_SVT,a,0.4211153812640536,0.8307522123893806,0.8825833123189902,0.6433566433566433
87
+ Twins_SVT,b,0.3625493723054758,0.8550770198050928,0.8962191528545118,0.37449118046132973
88
+ Twins_SVT,c,0.47319920195681764,0.7868594781515247,0.8548139963167587,0.2893081761006289
89
+ Twins_SVT,d,0.1203458983801289,0.9783087079534738,0.9818324125230202,0.8
90
+ Twins_SVT,e,0.5213294555274637,0.7486278814489572,0.8316203738742148,0.5465346534653466
91
+ Twins_SVT,f,0.3335461875583885,0.8666541282678202,0.9034523383543173,0.16292798110979928
92
+ Twins_SVT,g,0.2639119902451833,0.9085,0.9744078888888889,0.912676952441546
93
+ Twins_SVT,h,0.32257486327489215,0.8723333333333333,0.9662636666666669,0.8822263222632226
94
+ Twins_SVT,i,0.13550377811988196,0.9738333333333333,0.9972788888888889,0.9733672603901612
95
+ Twins_SVT,j,1.2430085968176523,0.49,0.43771377777777776,0.1896186440677966
96
+ Twins_SVT,k,1.1146003757913907,0.5553333333333333,0.7234002222222222,0.2115839243498818
97
+ Twins_SVT,l,0.6286477774643219,0.7480461704941685,0.7275090480198628,0.6162439337057046
98
+ Twins_PCPVT,a,0.45601994748664115,0.7699115044247787,0.8394007473464615,0.5458515283842795
99
+ Twins_PCPVT,b,0.3125818614145001,0.8773970449544168,0.9010699815837937,0.390625
100
+ Twins_PCPVT,c,0.5049686531944119,0.7500785916378497,0.8135911602209945,0.23923444976076555
101
+ Twins_PCPVT,d,0.3149096430453517,0.8918579063187677,0.9015690607734806,0.4208754208754209
102
+ Twins_PCPVT,e,0.42039827045572575,0.8079034028540066,0.8655339438431847,0.5882352941176471
103
+ Twins_PCPVT,f,0.3770137148085496,0.8412638706037239,0.8693597175042401,0.12899896800825594
104
+ Twins_PCPVT,g,0.2785677030881246,0.9015,0.9626754444444443,0.9027480664801711
105
+ Twins_PCPVT,h,0.3805647597312927,0.834,0.928301,0.8463437210737427
106
+ Twins_PCPVT,i,0.2798018006483714,0.9091666666666667,0.9656723333333334,0.9096335599403084
107
+ Twins_PCPVT,j,0.614702238559723,0.6835,0.7995154444444446,0.6018033130635353
108
+ Twins_PCPVT,k,0.6159363424777985,0.6911666666666667,0.7903985,0.6076646199449502
109
+ Twins_PCPVT,l,0.45535326129802217,0.7889864133702056,0.8498913163479216,0.7103004291845494
110
+ PiT,a,0.3937257931823224,0.8296460176991151,0.8874127904755356,0.641860465116279
111
+ PiT,b,0.2796248870145521,0.8777114115058158,0.91848802946593,0.4150375939849624
112
+ PiT,c,0.5313189482209218,0.7613957874882112,0.8498581952117863,0.26666666666666666
113
+ PiT,d,0.049343678185640734,0.9798805407104684,0.9911620626151012,0.8117647058823529
114
+ PiT,e,0.3259278782832505,0.8518111964873765,0.9145841216983274,0.6715328467153284
115
+ PiT,f,0.2841162405192056,0.8750235094978371,0.9172267022129574,0.17196261682242991
116
+ PiT,g,0.1590204114516576,0.9338333333333333,0.9916004444444445,0.9369340746624305
117
+ PiT,h,0.2924602138201396,0.8721666666666666,0.981646111111111,0.8849212303075769
118
+ PiT,i,0.03693298858900865,0.988,0.999485,0.9879396984924623
119
+ PiT,j,2.9977854507366817,0.461,0.277717,0.06477732793522267
120
+ PiT,k,2.8756980224698783,0.5151666666666667,0.7229978888888889,0.07149696776252792
121
+ PiT,l,1.2244331041709067,0.7434170975111218,0.6790239785353327,0.599849990624414
122
+ Ensemble,a,,0.9070796460176991,0.941851401847734,0.79
123
+ Ensemble,b,,0.9374410562716127,0.9600349907918969,0.6135922330097088
124
+ Ensemble,c,,0.895001571832757,0.9307624309392265,0.48615384615384616
125
+ Ensemble,d,,0.9911977365608299,0.9944677716390424,0.9186046511627907
126
+ Ensemble,e,,0.9264544456641054,0.955384848255506,0.825065274151436
127
+ Ensemble,f,,0.941696445363927,0.9599335198386041,0.33760683760683763
128
+ Ensemble,g,,0.9701666666666666,0.9990522222222222,0.9710027539283979
129
+ Ensemble,h,,0.9476666666666667,0.9979163333333333,0.9502219403931516
130
+ Ensemble,i,,0.9986666666666667,0.9999886666666667,0.9986671109630123
131
+ Ensemble,j,,0.5698333333333333,0.6426453333333333,0.31556616282153277
132
+ Ensemble,k,,0.5983333333333334,0.8897323333333333,0.33055555555555555
133
+ Ensemble,l,,0.8179632078874595,0.832089495815299,0.712386018237082
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32fb6c4540f07c780b611960417a6e5c764d63415407070bfb0afcddcd3dea7e
3
+ size 290985688
pit-gravit-c2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:610198bdc074942f3128395ac2af223bc5692124344e78135ce9e033f2540fab
3
+ size 291039882
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:610198bdc074942f3128395ac2af223bc5692124344e78135ce9e033f2540fab
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: 9e835ba6293ac0c0be4d9dd939a4de027b7225fcfc9438ddbb4b782f4cc7067b
  • Pointer size: 131 Bytes
  • Size of remote file: 158 kB
training_curves/PiT_f1.png ADDED
training_curves/PiT_loss.png ADDED
training_curves/PiT_metrics.csv ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1
2
+ 1,0.36233020514299746,0.29970101251894116,0.8383384068132845,0.8709912536443148,0.9196018811517546,0.9604002158964379,0.8387575478456657,0.88
3
+ 2,0.2789034600516703,0.2393716046744116,0.8829821801142388,0.9037900874635568,0.9529643326162407,0.9694663788047497,0.8823332158241833,0.9074333800841514
4
+ 3,0.2593451212059152,0.20805626486376494,0.8912850155624722,0.9198250728862973,0.9592965369511129,0.9765743865226224,0.8906585940590653,0.9217638691322901
5
+ 4,0.24514418407088653,0.19002231221331103,0.8979033416561206,0.923469387755102,0.9636186699065337,0.9771683142228154,0.8973609559013153,0.922509225092251
6
+ 5,0.2394631020257153,0.17069047450497965,0.8997845196155556,0.9336734693877551,0.9651783712893223,0.9818506744638714,0.8992451986726501,0.9324424647364514
7
+ 6,0.23080718357165805,0.24374222516492228,0.9044190580428909,0.8965014577259475,0.9676906088606333,0.9805076966230057,0.903995465164216,0.8865814696485623
8
+ 7,0.22450651501304095,0.17771336192987403,0.9064883537982693,0.9314868804664723,0.9693469276281166,0.9802898877168528,0.905988343104723,0.9321789321789322
9
+ 8,0.21979985898629778,0.17595275024457158,0.9086944624961522,0.934402332361516,0.9705851395657841,0.9828015962736614,0.9082661809934537,0.9356223175965666
10
+ 9,0.21516496794928433,0.1790100717727019,0.9112426035502958,0.9314868804664723,0.9718440235380357,0.9838619537777628,0.9108078846517383,0.9335219236209336
11
+ 10,0.21452437499363183,0.17893647717491185,0.9108663679584089,0.9285714285714286,0.9719583923985107,0.9853281795850368,0.9103219257041587,0.9247311827956989
12
+ 11,0.20754402267739014,0.17404167527881378,0.9134914662927113,0.9307580174927114,0.9737529269184622,0.9835400215896437,0.9130700028355144,0.931899641577061
13
+ 12,0.19780038125214403,0.1418822090584057,0.9193316687758663,0.9467930029154519,0.9763044930341069,0.987497768786815,0.9189658134341179,0.9460458240946046
14
+ 13,0.1934141220649967,0.14139472395616093,0.9206741457741903,0.9467930029154519,0.9773581883014619,0.9875806424194001,0.9202095177479422,0.9467541940189642
15
+ 14,0.1934208858461457,0.139229262144503,0.9201268940041728,0.9453352769679301,0.9773056264320266,0.9878207634574029,0.9197708474692731,0.9452154857560263
16
+ 15,0.19074950850326525,0.1402430695104182,0.922324451893149,0.9460641399416909,0.9779107379347325,0.9874095827418847,0.921859408870385,0.9455081001472754
17
+ 16,0.19173615694547533,0.13882749668884556,0.9207083490098164,0.9467930029154519,0.9776180775285328,0.9877835765709866,0.9202301994890191,0.9460458240946046
18
+ 17,0.191168249241059,0.13754117918952907,0.920964873277012,0.9453352769679301,0.9777927338764157,0.9879004496425808,0.9206030150753769,0.9448123620309051
19
+ 18,0.18873048595214878,0.13981756654842253,0.9223586551287751,0.9453352769679301,0.9783731073452622,0.9876592661221089,0.9219179966978536,0.9444855662472242
20
+ 19,0.19039970816815727,0.13393708690457892,0.9230683722680165,0.9511661807580175,0.9779171543316474,0.988471002728455,0.9227387096497179,0.9508437270726339
21
+ 20,0.1886889381845701,0.13384573848011194,0.9220422751992339,0.9467930029154519,0.9783297421944188,0.9887684978197859,0.9216099326758553,0.9469090909090909
22
+ 21,0.18851448938861634,0.13450837917300076,0.9226750350583165,0.9489795918367347,0.9784135740184345,0.9883286300776035,0.9222702619070131,0.948301329394387
23
+ 22,0.18751028372420084,0.1397644239681455,0.9226750350583165,0.9438775510204082,0.9785963369203852,0.9883350049724179,0.922244864618533,0.9444844989185291
24
+ 23,0.18757704842560316,0.13256130466351704,0.9234873619044361,0.9482507288629738,0.9785667280226568,0.9887908099516357,0.9230848576536927,0.9478324761204996
25
+ 24,0.18770277607335753,0.1357979845505414,0.9227263399117557,0.9460641399416909,0.9785498741229798,0.9884688777635169,0.9221974464714644,0.9460641399416909
26
+ 25,0.1863581392493963,0.1353398425933571,0.9238037418339775,0.9482507288629738,0.9788793404137933,0.9885124395447475,0.9234364662720064,0.9483636363636364
27
+ 26,0.18420615110185887,0.14155046685270248,0.9249837534630776,0.9424198250728864,0.9793952761659406,0.9878866373704833,0.924578099880501,0.9428778018799711
28
+ 27,0.18581719516048345,0.13698511442533387,0.9237267845538187,0.9467930029154519,0.9789779452466503,0.9884996897551189,0.9234058630579265,0.9470630891950689
29
+ 28,0.18425983067184118,0.13252408906773994,0.9248811437561993,0.9475218658892128,0.9793347893634089,0.9887865600217596,0.9244580499256189,0.947136563876652
30
+ 29,0.18495915091218784,0.13212353425540313,0.9241115709546123,0.9467930029154519,0.9791665277700061,0.9892391775535703,0.9236998890961768,0.9469090909090909
31
+ 30,0.18430970521423262,0.12890243684498284,0.9239833088210144,0.9511661807580175,0.9793424826639624,0.9893231136686245,0.9236280540187622,0.9506990434142752
32
+ 31,0.1838982511615914,0.13194029050561498,0.924940999418545,0.9482507288629738,0.9794280160512789,0.9888630587595305,0.9246083550913838,0.9475240206947524
33
+ 32,0.18290547842928384,0.13082542394236296,0.9243253411772754,0.9504373177842566,0.979600852387402,0.9890351809195148,0.9239494715132767,0.9501466275659824
34
+ 33,0.18369765012185232,0.12840869347225473,0.9252744809658994,0.9533527696793003,0.9794289731440773,0.9894197995733071,0.9248587716356695,0.9528718703976435
35
+ 34,0.18320441776851862,0.13257058870601238,0.9256592673666929,0.9460641399416909,0.9795751438073387,0.9891520539911092,0.9252733273739944,0.9462209302325582
36
+ 35,0.17985989359902607,0.13369113070016003,0.9266768136265691,0.9467930029154519,0.9803058469426976,0.9887408732755909,0.92638031542708,0.9469862018881626
37
+ 36,0.18242066704542312,0.1306314041333018,0.9257362246468516,0.9497084548104956,0.9797273287276838,0.9891849909476493,0.9253538921692495,0.9498181818181818
38
+ 37,0.18303082494583306,0.13051823008555713,0.9251804220679276,0.9518950437317785,0.9795625560994569,0.989337988423191,0.9248423837419044,0.9511111111111111
39
+ 38,0.18136139805610413,0.12837442981605976,0.9255652084687211,0.9518950437317785,0.9799693281428723,0.9896046715229199,0.9251562647774463,0.9512555391432792
40
+ 39,0.18148975201810144,0.1306429403907356,0.9260697061942059,0.9467930029154519,0.9799571759158697,0.9891446166138258,0.9257305822323776,0.9461254612546125
41
+ 40,0.17934676769857233,0.14537533867897862,0.9271129048808017,0.9409620991253644,0.9804232608059748,0.9883530671743915,0.9267886283603882,0.9420186113099499
42
+ 41,0.17940764591583544,0.12568791645087585,0.9262407223723365,0.9518950437317785,0.9803314559383155,0.9898331052537633,0.9258794617539398,0.9516837481698389
43
+ 42,0.18107056629153379,0.13271132343972739,0.9263176796524951,0.9467930029154519,0.9799676722041636,0.9894219245382452,0.9259459784635745,0.9472924187725632
44
+ 43,0.17952147694171752,0.12523162727713932,0.9267024660532887,0.9518950437317785,0.9803587872622723,0.989962728114986,0.9263282740601956,0.9516129032258065
45
+ 44,0.1808621069753358,0.12804140719658433,0.926642610390943,0.9511661807580175,0.980039486042764,0.9895685471189726,0.9262674791367649,0.9509875640087784
46
+ 45,0.17865849226533778,0.12606683558086612,0.9270958032629887,0.9504373177842566,0.9805542058702942,0.9897130447347618,0.9267122816668959,0.9502196193265008
47
+ 46,0.17709277664536216,0.12949952900235925,0.9284040770256866,0.9518950437317785,0.9808630989798061,0.989336925940722,0.9280750431652822,0.9518950437317785
48
+ 47,0.1776126382921709,0.13869572819669462,0.9277285631220713,0.9453352769679301,0.9807427155697823,0.9896365459969911,0.9273496192129829,0.9463902787705504
49
+ 48,0.17726845076405065,0.1274495993254831,0.9266853644354756,0.9518950437317785,0.9808809555968733,0.9895111730656443,0.9263110851367379,0.9516837481698389
50
+ 49,0.17738705108885555,0.1285623211858919,0.9273694291479974,0.9489795918367347,0.9808116394122064,0.9895122355481135,0.9270700965071951,0.9489795918367347
51
+ 50,0.17664882755782685,0.1259572171435064,0.92741218319253,0.9540816326530612,0.9809636039599241,0.989799105814754,0.9269819970926982,0.9539810080350621
52
+ 51,0.17600744828802498,0.12716585508501566,0.9278055204022301,0.9497084548104956,0.9811120241202851,0.9897969808498159,0.9273901564340938,0.9498910675381264
53
+ 52,0.17542210658560187,0.12426498649592997,0.9281390019495844,0.9540816326530612,0.9812825283378522,0.990067913879421,0.9276764199655766,0.9539810080350621
54
+ 53,0.1766666979093038,0.12476968452464735,0.9278568252556691,0.9533527696793003,0.9810123785495447,0.9900339144404118,0.9274554818961144,0.9532846715328467
55
+ 54,0.17660799297673369,0.126970670748035,0.9284981359236584,0.9504373177842566,0.9809859233054894,0.9900456017475712,0.9280849014413981,0.9506531204644412
56
+ 55,0.17558396369189266,0.12640303398316874,0.9275318945172213,0.9504373177842566,0.9812259422916837,0.9899691030098002,0.9271362616388538,0.9505813953488372
57
+ 56,0.17395585639141128,0.1250900527565542,0.9286263980572562,0.9526239067055393,0.9815714387535673,0.9900572890547306,0.9282897619395356,0.9526584122359796
58
+ 57,0.17536393163912592,0.12370957225648029,0.9280962479050519,0.9533527696793003,0.9812844947285105,0.9901571624068203,0.9277074252701623,0.9532846715328467
59
+ 58,0.17566584592891074,0.12333806076306991,0.9286092964394431,0.9526239067055393,0.9811826397284211,0.9901699121964488,0.9281899109792285,0.9524506217995611
60
+ 59,0.17558764315387043,0.1256001479163462,0.9283527721722475,0.9526239067055393,0.9812124197183688,0.98995847818511,0.9279839104762396,0.9525893508388038
61
+ 60,0.17575680659326715,0.12565061119765195,0.9283100181277149,0.9518950437317785,0.9811708919809725,0.9900307269930045,0.9279762211569851,0.9518950437317785
62
+ 61,0.17540902452454887,0.12461533680905754,0.9288059650442931,0.9518950437317785,0.9812550357192646,0.9901051007658375,0.9284179032618601,0.9518950437317785
63
+ 62,0.17541102996259683,0.12432444899318518,0.9282245100386497,0.9533527696793003,0.9812681197408146,0.990126350415218,0.9278270738753611,0.9532846715328467
64
+ 63,0.1749994034859829,0.12393498219245551,0.9283100181277149,0.9533527696793003,0.9813248757093408,0.990092350976209,0.9279527017736835,0.9532846715328467
65
+ 64,0.17485610548442898,0.12386349512606251,0.9280620446694258,0.9533527696793003,0.9813954807887251,0.9902049741179271,0.9277215048497813,0.9532846715328467
66
+ 65,0.1759885422194563,0.12439083941172233,0.92741218319253,0.9533527696793003,0.9811308063901275,0.9901656622665727,0.9270810964034463,0.9532846715328467
67
+ 66,0.1759207148018135,0.12433676582383693,0.9277542155487909,0.9533527696793003,0.9811570319026647,0.9901645997841035,0.9273473037930056,0.9532846715328467
68
+ 67,0.17424281581552825,0.12447478372397297,0.9291736498272737,0.9533527696793003,0.9815324444971079,0.9901242254502801,0.9287453223794572,0.9532846715328467
69
+ 68,0.17385396183986573,0.12418095758809293,0.9288914731333584,0.9533527696793003,0.9815897931466292,0.9901380377223776,0.9286033174216147,0.9532846715328467
70
+ 69,0.1750469607995702,0.12394943818145869,0.9287119061463215,0.9533527696793003,0.9813509181385803,0.9901422876522538,0.9283941286105696,0.9532846715328467
71
+ 70,0.17642186737588156,0.12356808784882112,0.9287888634264801,0.9533527696793003,0.9810236544041807,0.9901922243282987,0.9284695858313435,0.9532846715328467
72
+ 71,0.17437013022288922,0.12374842514449584,0.9286007456305366,0.9533527696793003,0.9815056148974278,0.9901752246087939,0.928130002926443,0.9532846715328467
73
+ 72,0.1753760602373747,0.12390452723287633,0.9293446660054041,0.9533527696793003,0.9812637434358219,0.9901667247490417,0.9289026940053863,0.9532846715328467
74
+ 73,0.17607059463969202,0.12404029848971798,0.9281732051852105,0.9533527696793003,0.9811083219478821,0.9901656622665727,0.9277854195323246,0.9532846715328467
75
+ 74,0.17546011868152164,0.12413608482645135,0.9286434996750692,0.9533527696793003,0.9812536699061651,0.9901624748191655,0.9282440647652132,0.9532846715328467
76
+ 75,0.17476058825221047,0.12416948035424548,0.9283356705544344,0.9533527696793003,0.9814359636937233,0.9901592873717584,0.9279109574312526,0.9532846715328467
77
+ 76,0.1753198992475364,0.12414770433271939,0.9284639326880323,0.9533527696793003,0.9812443771127117,0.9901614123366964,0.9280950252690205,0.9532846715328467
78
+ 77,0.1746743339889128,0.12415588093221014,0.9289085747511715,0.9533527696793003,0.9814541446548435,0.9901603498542274,0.9285087794727157,0.9532846715328467
79
+ 78,0.17683591671265333,0.12414996087029793,0.9266084071553169,0.9533527696793003,0.9809634792234599,0.9901635373016345,0.9262951799469304,0.9532846715328467
training_metrics.csv ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1
2
+ 1,0.36233020514299746,0.29970101251894116,0.8383384068132845,0.8709912536443148,0.9196018811517546,0.9604002158964379,0.8387575478456657,0.88
3
+ 2,0.2789034600516703,0.2393716046744116,0.8829821801142388,0.9037900874635568,0.9529643326162407,0.9694663788047497,0.8823332158241833,0.9074333800841514
4
+ 3,0.2593451212059152,0.20805626486376494,0.8912850155624722,0.9198250728862973,0.9592965369511129,0.9765743865226224,0.8906585940590653,0.9217638691322901
5
+ 4,0.24514418407088653,0.19002231221331103,0.8979033416561206,0.923469387755102,0.9636186699065337,0.9771683142228154,0.8973609559013153,0.922509225092251
6
+ 5,0.2394631020257153,0.17069047450497965,0.8997845196155556,0.9336734693877551,0.9651783712893223,0.9818506744638714,0.8992451986726501,0.9324424647364514
7
+ 6,0.23080718357165805,0.24374222516492228,0.9044190580428909,0.8965014577259475,0.9676906088606333,0.9805076966230057,0.903995465164216,0.8865814696485623
8
+ 7,0.22450651501304095,0.17771336192987403,0.9064883537982693,0.9314868804664723,0.9693469276281166,0.9802898877168528,0.905988343104723,0.9321789321789322
9
+ 8,0.21979985898629778,0.17595275024457158,0.9086944624961522,0.934402332361516,0.9705851395657841,0.9828015962736614,0.9082661809934537,0.9356223175965666
10
+ 9,0.21516496794928433,0.1790100717727019,0.9112426035502958,0.9314868804664723,0.9718440235380357,0.9838619537777628,0.9108078846517383,0.9335219236209336
11
+ 10,0.21452437499363183,0.17893647717491185,0.9108663679584089,0.9285714285714286,0.9719583923985107,0.9853281795850368,0.9103219257041587,0.9247311827956989
12
+ 11,0.20754402267739014,0.17404167527881378,0.9134914662927113,0.9307580174927114,0.9737529269184622,0.9835400215896437,0.9130700028355144,0.931899641577061
13
+ 12,0.19780038125214403,0.1418822090584057,0.9193316687758663,0.9467930029154519,0.9763044930341069,0.987497768786815,0.9189658134341179,0.9460458240946046
14
+ 13,0.1934141220649967,0.14139472395616093,0.9206741457741903,0.9467930029154519,0.9773581883014619,0.9875806424194001,0.9202095177479422,0.9467541940189642
15
+ 14,0.1934208858461457,0.139229262144503,0.9201268940041728,0.9453352769679301,0.9773056264320266,0.9878207634574029,0.9197708474692731,0.9452154857560263
16
+ 15,0.19074950850326525,0.1402430695104182,0.922324451893149,0.9460641399416909,0.9779107379347325,0.9874095827418847,0.921859408870385,0.9455081001472754
17
+ 16,0.19173615694547533,0.13882749668884556,0.9207083490098164,0.9467930029154519,0.9776180775285328,0.9877835765709866,0.9202301994890191,0.9460458240946046
18
+ 17,0.191168249241059,0.13754117918952907,0.920964873277012,0.9453352769679301,0.9777927338764157,0.9879004496425808,0.9206030150753769,0.9448123620309051
19
+ 18,0.18873048595214878,0.13981756654842253,0.9223586551287751,0.9453352769679301,0.9783731073452622,0.9876592661221089,0.9219179966978536,0.9444855662472242
20
+ 19,0.19039970816815727,0.13393708690457892,0.9230683722680165,0.9511661807580175,0.9779171543316474,0.988471002728455,0.9227387096497179,0.9508437270726339
21
+ 20,0.1886889381845701,0.13384573848011194,0.9220422751992339,0.9467930029154519,0.9783297421944188,0.9887684978197859,0.9216099326758553,0.9469090909090909
22
+ 21,0.18851448938861634,0.13450837917300076,0.9226750350583165,0.9489795918367347,0.9784135740184345,0.9883286300776035,0.9222702619070131,0.948301329394387
23
+ 22,0.18751028372420084,0.1397644239681455,0.9226750350583165,0.9438775510204082,0.9785963369203852,0.9883350049724179,0.922244864618533,0.9444844989185291
24
+ 23,0.18757704842560316,0.13256130466351704,0.9234873619044361,0.9482507288629738,0.9785667280226568,0.9887908099516357,0.9230848576536927,0.9478324761204996
25
+ 24,0.18770277607335753,0.1357979845505414,0.9227263399117557,0.9460641399416909,0.9785498741229798,0.9884688777635169,0.9221974464714644,0.9460641399416909
26
+ 25,0.1863581392493963,0.1353398425933571,0.9238037418339775,0.9482507288629738,0.9788793404137933,0.9885124395447475,0.9234364662720064,0.9483636363636364
27
+ 26,0.18420615110185887,0.14155046685270248,0.9249837534630776,0.9424198250728864,0.9793952761659406,0.9878866373704833,0.924578099880501,0.9428778018799711
28
+ 27,0.18581719516048345,0.13698511442533387,0.9237267845538187,0.9467930029154519,0.9789779452466503,0.9884996897551189,0.9234058630579265,0.9470630891950689
29
+ 28,0.18425983067184118,0.13252408906773994,0.9248811437561993,0.9475218658892128,0.9793347893634089,0.9887865600217596,0.9244580499256189,0.947136563876652
30
+ 29,0.18495915091218784,0.13212353425540313,0.9241115709546123,0.9467930029154519,0.9791665277700061,0.9892391775535703,0.9236998890961768,0.9469090909090909
31
+ 30,0.18430970521423262,0.12890243684498284,0.9239833088210144,0.9511661807580175,0.9793424826639624,0.9893231136686245,0.9236280540187622,0.9506990434142752
32
+ 31,0.1838982511615914,0.13194029050561498,0.924940999418545,0.9482507288629738,0.9794280160512789,0.9888630587595305,0.9246083550913838,0.9475240206947524
33
+ 32,0.18290547842928384,0.13082542394236296,0.9243253411772754,0.9504373177842566,0.979600852387402,0.9890351809195148,0.9239494715132767,0.9501466275659824
34
+ 33,0.18369765012185232,0.12840869347225473,0.9252744809658994,0.9533527696793003,0.9794289731440773,0.9894197995733071,0.9248587716356695,0.9528718703976435
35
+ 34,0.18320441776851862,0.13257058870601238,0.9256592673666929,0.9460641399416909,0.9795751438073387,0.9891520539911092,0.9252733273739944,0.9462209302325582
36
+ 35,0.17985989359902607,0.13369113070016003,0.9266768136265691,0.9467930029154519,0.9803058469426976,0.9887408732755909,0.92638031542708,0.9469862018881626
37
+ 36,0.18242066704542312,0.1306314041333018,0.9257362246468516,0.9497084548104956,0.9797273287276838,0.9891849909476493,0.9253538921692495,0.9498181818181818
38
+ 37,0.18303082494583306,0.13051823008555713,0.9251804220679276,0.9518950437317785,0.9795625560994569,0.989337988423191,0.9248423837419044,0.9511111111111111
39
+ 38,0.18136139805610413,0.12837442981605976,0.9255652084687211,0.9518950437317785,0.9799693281428723,0.9896046715229199,0.9251562647774463,0.9512555391432792
40
+ 39,0.18148975201810144,0.1306429403907356,0.9260697061942059,0.9467930029154519,0.9799571759158697,0.9891446166138258,0.9257305822323776,0.9461254612546125
41
+ 40,0.17934676769857233,0.14537533867897862,0.9271129048808017,0.9409620991253644,0.9804232608059748,0.9883530671743915,0.9267886283603882,0.9420186113099499
42
+ 41,0.17940764591583544,0.12568791645087585,0.9262407223723365,0.9518950437317785,0.9803314559383155,0.9898331052537633,0.9258794617539398,0.9516837481698389
43
+ 42,0.18107056629153379,0.13271132343972739,0.9263176796524951,0.9467930029154519,0.9799676722041636,0.9894219245382452,0.9259459784635745,0.9472924187725632
44
+ 43,0.17952147694171752,0.12523162727713932,0.9267024660532887,0.9518950437317785,0.9803587872622723,0.989962728114986,0.9263282740601956,0.9516129032258065
45
+ 44,0.1808621069753358,0.12804140719658433,0.926642610390943,0.9511661807580175,0.980039486042764,0.9895685471189726,0.9262674791367649,0.9509875640087784
46
+ 45,0.17865849226533778,0.12606683558086612,0.9270958032629887,0.9504373177842566,0.9805542058702942,0.9897130447347618,0.9267122816668959,0.9502196193265008
47
+ 46,0.17709277664536216,0.12949952900235925,0.9284040770256866,0.9518950437317785,0.9808630989798061,0.989336925940722,0.9280750431652822,0.9518950437317785
48
+ 47,0.1776126382921709,0.13869572819669462,0.9277285631220713,0.9453352769679301,0.9807427155697823,0.9896365459969911,0.9273496192129829,0.9463902787705504
49
+ 48,0.17726845076405065,0.1274495993254831,0.9266853644354756,0.9518950437317785,0.9808809555968733,0.9895111730656443,0.9263110851367379,0.9516837481698389
50
+ 49,0.17738705108885555,0.1285623211858919,0.9273694291479974,0.9489795918367347,0.9808116394122064,0.9895122355481135,0.9270700965071951,0.9489795918367347
51
+ 50,0.17664882755782685,0.1259572171435064,0.92741218319253,0.9540816326530612,0.9809636039599241,0.989799105814754,0.9269819970926982,0.9539810080350621
52
+ 51,0.17600744828802498,0.12716585508501566,0.9278055204022301,0.9497084548104956,0.9811120241202851,0.9897969808498159,0.9273901564340938,0.9498910675381264
53
+ 52,0.17542210658560187,0.12426498649592997,0.9281390019495844,0.9540816326530612,0.9812825283378522,0.990067913879421,0.9276764199655766,0.9539810080350621
54
+ 53,0.1766666979093038,0.12476968452464735,0.9278568252556691,0.9533527696793003,0.9810123785495447,0.9900339144404118,0.9274554818961144,0.9532846715328467
55
+ 54,0.17660799297673369,0.126970670748035,0.9284981359236584,0.9504373177842566,0.9809859233054894,0.9900456017475712,0.9280849014413981,0.9506531204644412
56
+ 55,0.17558396369189266,0.12640303398316874,0.9275318945172213,0.9504373177842566,0.9812259422916837,0.9899691030098002,0.9271362616388538,0.9505813953488372
57
+ 56,0.17395585639141128,0.1250900527565542,0.9286263980572562,0.9526239067055393,0.9815714387535673,0.9900572890547306,0.9282897619395356,0.9526584122359796
58
+ 57,0.17536393163912592,0.12370957225648029,0.9280962479050519,0.9533527696793003,0.9812844947285105,0.9901571624068203,0.9277074252701623,0.9532846715328467
59
+ 58,0.17566584592891074,0.12333806076306991,0.9286092964394431,0.9526239067055393,0.9811826397284211,0.9901699121964488,0.9281899109792285,0.9524506217995611
60
+ 59,0.17558764315387043,0.1256001479163462,0.9283527721722475,0.9526239067055393,0.9812124197183688,0.98995847818511,0.9279839104762396,0.9525893508388038
61
+ 60,0.17575680659326715,0.12565061119765195,0.9283100181277149,0.9518950437317785,0.9811708919809725,0.9900307269930045,0.9279762211569851,0.9518950437317785
62
+ 61,0.17540902452454887,0.12461533680905754,0.9288059650442931,0.9518950437317785,0.9812550357192646,0.9901051007658375,0.9284179032618601,0.9518950437317785
63
+ 62,0.17541102996259683,0.12432444899318518,0.9282245100386497,0.9533527696793003,0.9812681197408146,0.990126350415218,0.9278270738753611,0.9532846715328467
64
+ 63,0.1749994034859829,0.12393498219245551,0.9283100181277149,0.9533527696793003,0.9813248757093408,0.990092350976209,0.9279527017736835,0.9532846715328467
65
+ 64,0.17485610548442898,0.12386349512606251,0.9280620446694258,0.9533527696793003,0.9813954807887251,0.9902049741179271,0.9277215048497813,0.9532846715328467
66
+ 65,0.1759885422194563,0.12439083941172233,0.92741218319253,0.9533527696793003,0.9811308063901275,0.9901656622665727,0.9270810964034463,0.9532846715328467
67
+ 66,0.1759207148018135,0.12433676582383693,0.9277542155487909,0.9533527696793003,0.9811570319026647,0.9901645997841035,0.9273473037930056,0.9532846715328467
68
+ 67,0.17424281581552825,0.12447478372397297,0.9291736498272737,0.9533527696793003,0.9815324444971079,0.9901242254502801,0.9287453223794572,0.9532846715328467
69
+ 68,0.17385396183986573,0.12418095758809293,0.9288914731333584,0.9533527696793003,0.9815897931466292,0.9901380377223776,0.9286033174216147,0.9532846715328467
70
+ 69,0.1750469607995702,0.12394943818145869,0.9287119061463215,0.9533527696793003,0.9813509181385803,0.9901422876522538,0.9283941286105696,0.9532846715328467
71
+ 70,0.17642186737588156,0.12356808784882112,0.9287888634264801,0.9533527696793003,0.9810236544041807,0.9901922243282987,0.9284695858313435,0.9532846715328467
72
+ 71,0.17437013022288922,0.12374842514449584,0.9286007456305366,0.9533527696793003,0.9815056148974278,0.9901752246087939,0.928130002926443,0.9532846715328467
73
+ 72,0.1753760602373747,0.12390452723287633,0.9293446660054041,0.9533527696793003,0.9812637434358219,0.9901667247490417,0.9289026940053863,0.9532846715328467
74
+ 73,0.17607059463969202,0.12404029848971798,0.9281732051852105,0.9533527696793003,0.9811083219478821,0.9901656622665727,0.9277854195323246,0.9532846715328467
75
+ 74,0.17546011868152164,0.12413608482645135,0.9286434996750692,0.9533527696793003,0.9812536699061651,0.9901624748191655,0.9282440647652132,0.9532846715328467
76
+ 75,0.17476058825221047,0.12416948035424548,0.9283356705544344,0.9533527696793003,0.9814359636937233,0.9901592873717584,0.9279109574312526,0.9532846715328467
77
+ 76,0.1753198992475364,0.12414770433271939,0.9284639326880323,0.9533527696793003,0.9812443771127117,0.9901614123366964,0.9280950252690205,0.9532846715328467
78
+ 77,0.1746743339889128,0.12415588093221014,0.9289085747511715,0.9533527696793003,0.9814541446548435,0.9901603498542274,0.9285087794727157,0.9532846715328467
79
+ 78,0.17683591671265333,0.12414996087029793,0.9266084071553169,0.9533527696793003,0.9809634792234599,0.9901635373016345,0.9262951799469304,0.9532846715328467