parlange commited on
Commit
038dced
·
verified ·
1 Parent(s): 3df7312

Upload DeiT3 model from experiment a1

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/DeiT3_Confusion_Matrix_a.png +0 -0
  5. confusion_matrices/DeiT3_Confusion_Matrix_b.png +0 -0
  6. confusion_matrices/DeiT3_Confusion_Matrix_c.png +0 -0
  7. confusion_matrices/DeiT3_Confusion_Matrix_d.png +0 -0
  8. confusion_matrices/DeiT3_Confusion_Matrix_e.png +0 -0
  9. confusion_matrices/DeiT3_Confusion_Matrix_f.png +0 -0
  10. confusion_matrices/DeiT3_Confusion_Matrix_g.png +0 -0
  11. confusion_matrices/DeiT3_Confusion_Matrix_h.png +0 -0
  12. confusion_matrices/DeiT3_Confusion_Matrix_i.png +0 -0
  13. confusion_matrices/DeiT3_Confusion_Matrix_j.png +0 -0
  14. confusion_matrices/DeiT3_Confusion_Matrix_k.png +0 -0
  15. confusion_matrices/DeiT3_Confusion_Matrix_l.png +0 -0
  16. deit3-gravit-a1.pth +3 -0
  17. evaluation_results.csv +133 -0
  18. model.safetensors +3 -0
  19. pytorch_model.bin +3 -0
  20. roc_confusion_matrix/DeiT3_roc_confusion_matrix_a.png +0 -0
  21. roc_confusion_matrix/DeiT3_roc_confusion_matrix_b.png +0 -0
  22. roc_confusion_matrix/DeiT3_roc_confusion_matrix_c.png +0 -0
  23. roc_confusion_matrix/DeiT3_roc_confusion_matrix_d.png +0 -0
  24. roc_confusion_matrix/DeiT3_roc_confusion_matrix_e.png +0 -0
  25. roc_confusion_matrix/DeiT3_roc_confusion_matrix_f.png +0 -0
  26. roc_confusion_matrix/DeiT3_roc_confusion_matrix_g.png +0 -0
  27. roc_confusion_matrix/DeiT3_roc_confusion_matrix_h.png +0 -0
  28. roc_confusion_matrix/DeiT3_roc_confusion_matrix_i.png +0 -0
  29. roc_confusion_matrix/DeiT3_roc_confusion_matrix_j.png +0 -0
  30. roc_confusion_matrix/DeiT3_roc_confusion_matrix_k.png +0 -0
  31. roc_confusion_matrix/DeiT3_roc_confusion_matrix_l.png +0 -0
  32. roc_curves/DeiT3_ROC_a.png +0 -0
  33. roc_curves/DeiT3_ROC_b.png +0 -0
  34. roc_curves/DeiT3_ROC_c.png +0 -0
  35. roc_curves/DeiT3_ROC_d.png +0 -0
  36. roc_curves/DeiT3_ROC_e.png +0 -0
  37. roc_curves/DeiT3_ROC_f.png +0 -0
  38. roc_curves/DeiT3_ROC_g.png +0 -0
  39. roc_curves/DeiT3_ROC_h.png +0 -0
  40. roc_curves/DeiT3_ROC_i.png +0 -0
  41. roc_curves/DeiT3_ROC_j.png +0 -0
  42. roc_curves/DeiT3_ROC_k.png +0 -0
  43. roc_curves/DeiT3_ROC_l.png +0 -0
  44. training_curves/DeiT3_accuracy.png +0 -0
  45. training_curves/DeiT3_auc.png +0 -0
  46. training_curves/DeiT3_combined_metrics.png +3 -0
  47. training_curves/DeiT3_f1.png +0 -0
  48. training_curves/DeiT3_loss.png +0 -0
  49. training_curves/DeiT3_metrics.csv +101 -0
  50. training_metrics.csv +101 -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/DeiT3_combined_metrics.png filter=lfs diff=lfs merge=lfs -text
37
+ training_notebook_a1.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
+ - deit3
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: DeiT3-a1
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.8213
31
+ name: Average Accuracy
32
+ - type: auc
33
+ value: 0.8238
34
+ name: Average AUC-ROC
35
+ - type: f1
36
+ value: 0.5164
37
+ name: Average F1-Score
38
+ ---
39
+
40
+ # 🌌 deit3-gravit-a1
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**: DeiT3
49
+ - **🧪 Experiment**: A1 - C21-classification-head
50
+ - **🌌 Dataset**: C21
51
+ - **🪐 Fine-tuning Strategy**: classification-head
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/deit3-gravit-a1',
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:** classification-head
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 | classification_head |
92
+ | Stochastic Depth Probability | 0.1 |
93
+
94
+
95
+ ## 📈 Training Curves
96
+
97
+ ![Combined Training Metrics](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/training_curves/DeiT3_combined_metrics.png)
98
+
99
+
100
+ ## 🏁 Final Epoch Training Metrics
101
+
102
+ | Metric | Training | Validation |
103
+ |:---------:|:-----------:|:-------------:|
104
+ | 📉 Loss | 0.2031 | 0.2811 |
105
+ | 🎯 Accuracy | 0.9194 | 0.9070 |
106
+ | 📊 AUC-ROC | 0.9759 | 0.9564 |
107
+ | ⚖️ F1 Score | 0.9194 | 0.9071 |
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/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_a.png)
117
+ ![ROC + Confusion Matrix - Dataset B](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_b.png)
118
+ ![ROC + Confusion Matrix - Dataset C](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_c.png)
119
+ ![ROC + Confusion Matrix - Dataset D](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_d.png)
120
+ ![ROC + Confusion Matrix - Dataset E](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_e.png)
121
+ ![ROC + Confusion Matrix - Dataset F](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_f.png)
122
+ ![ROC + Confusion Matrix - Dataset G](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_g.png)
123
+ ![ROC + Confusion Matrix - Dataset H](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_h.png)
124
+ ![ROC + Confusion Matrix - Dataset I](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_i.png)
125
+ ![ROC + Confusion Matrix - Dataset J](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_j.png)
126
+ ![ROC + Confusion Matrix - Dataset K](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_roc_confusion_matrix_k.png)
127
+ ![ROC + Confusion Matrix - Dataset L](https://huggingface.co/parlange/deit3-gravit-a1/resolve/main/roc_confusion_matrix/DeiT3_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.8213 |
136
+ | 📈 Average AUC-ROC | 0.8238 |
137
+ | ⚖️ Average F1-Score | 0.5164 |
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": "deit3_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_a1",
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": "deit3_gravit_a1",
60
+ "experiment": "a1",
61
+ "training_strategy": "classification-head",
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": "classification_head",
72
+ "stochastic_depth_probability": "0.1"
73
+ },
74
+ "hf_hub_id": "parlange/deit3-gravit-a1",
75
+ "license": "apache-2.0"
76
+ }
confusion_matrices/DeiT3_Confusion_Matrix_a.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_b.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_c.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_d.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_e.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_f.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_g.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_h.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_i.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_j.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_k.png ADDED
confusion_matrices/DeiT3_Confusion_Matrix_l.png ADDED
deit3-gravit-a1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40c54c3c6d7bc946e7b20ade559330bb0c0e87e1a25101c0a457ffd93f6ff9d2
3
+ size 343337390
evaluation_results.csv ADDED
@@ -0,0 +1,133 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Model,Dataset,Loss,Accuracy,AUCROC,F1
2
+ ViT,a,0.2907674585365717,0.8663942156554543,0.8790764272559853,0.36281859070464767
3
+ ViT,b,0.3363787103321372,0.8409305249921408,0.8598268876611419,0.3235294117647059
4
+ ViT,c,0.3624865817540249,0.8145237346746306,0.8460092081031308,0.29086538461538464
5
+ ViT,d,0.20147570416317237,0.9100911662999057,0.914537753222836,0.4583333333333333
6
+ ViT,e,0.38351454359247183,0.8397365532381997,0.8551653674411563,0.6237113402061856
7
+ ViT,f,0.28049477608445245,0.8673224382309659,0.873733035896412,0.12378516624040921
8
+ ViT,g,0.20559095215797424,0.9115,0.982819,0.9165225593460148
9
+ ViT,h,0.21943247628211976,0.8975,0.9817622222222222,0.9045771916214119
10
+ ViT,i,0.13406987351179123,0.9481666666666667,0.9929653333333334,0.9493567822830158
11
+ ViT,j,1.5079676084518432,0.4855,0.5106800555555556,0.18869908015768724
12
+ ViT,k,1.4364465267062188,0.5221666666666667,0.6366698888888889,0.200278940027894
13
+ ViT,l,0.6356719980429736,0.7652688911215695,0.7653929261211886,0.6046842995814409
14
+ MLP-Mixer,a,0.23711907013393504,0.904118201823326,0.8973158379373849,0.40545808966861596
15
+ MLP-Mixer,b,0.32535140213071456,0.8626218170386671,0.8485451197053407,0.32248062015503876
16
+ MLP-Mixer,c,0.26777226502923085,0.8937441056271612,0.8793204419889503,0.38095238095238093
17
+ MLP-Mixer,d,0.24414238826503773,0.9034894687205282,0.8925340699815838,0.40388349514563104
18
+ MLP-Mixer,e,0.4684625132219982,0.7826564215148188,0.7941497010519943,0.5123152709359606
19
+ MLP-Mixer,f,0.25652938471803655,0.9010920920145612,0.8745385460021788,0.14006734006734006
20
+ MLP-Mixer,g,0.21773884654045106,0.9175,0.981478,0.9204819277108434
21
+ MLP-Mixer,h,0.18721229648590088,0.934,0.987164111111111,0.9353574926542605
22
+ MLP-Mixer,i,0.17468452334403992,0.9391666666666667,0.9881612222222221,0.9401148482362592
23
+ MLP-Mixer,j,1.2574891588687898,0.499,0.5052343333333333,0.19063004846526657
24
+ MLP-Mixer,k,1.2144348313808442,0.5206666666666667,0.6023342222222222,0.19754464285714285
25
+ MLP-Mixer,l,0.5478135314681644,0.7854158955105495,0.7718881621999252,0.6208893871449925
26
+ CvT,a,0.48862357091431496,0.7661112857591952,0.7645202578268876,0.23613963039014374
27
+ CvT,b,0.606147515758783,0.6821754165356806,0.7114585635359115,0.185334407735697
28
+ CvT,c,0.5599785684152655,0.7161270040867652,0.7296648250460406,0.2030008826125331
29
+ CvT,d,0.2567151319639595,0.8965734045897517,0.8801289134438306,0.41144901610017887
30
+ CvT,e,0.634658475738718,0.677277716794731,0.7027472943313403,0.4389312977099237
31
+ CvT,f,0.4719573438800043,0.7663232902176439,0.767503786678703,0.07083461656914075
32
+ CvT,g,0.37052693724632263,0.824,0.9481419444444444,0.845478489903424
33
+ CvT,h,0.34604967188835145,0.842,0.9550463333333333,0.8590544157002676
34
+ CvT,i,0.18526951444149017,0.9376666666666666,0.9862493333333333,0.9392067620286085
35
+ CvT,j,1.7297569365501404,0.37716666666666665,0.23870022222222223,0.10016855285335902
36
+ CvT,k,1.5444995085000992,0.49083333333333334,0.47829777777777777,0.1198501872659176
37
+ CvT,l,0.7996079008316138,0.686954682459944,0.637911609367734,0.5204147764095917
38
+ Swin,a,0.41406949968579354,0.8063502043382584,0.9315,0.346072186836518
39
+ Swin,b,0.46100307634250803,0.7742848160955674,0.9219889502762431,0.31226053639846746
40
+ Swin,c,0.4275265830337383,0.7950330084878969,0.9307624309392264,0.3333333333333333
41
+ Swin,d,0.12031445050063429,0.9597610814209369,0.9857863720073665,0.7180616740088106
42
+ Swin,e,0.7676329288210487,0.6114160263446762,0.8420797699235602,0.47941176470588237
43
+ Swin,f,0.39017997524092174,0.8144218108589575,0.9367503135673768,0.11976487876561352
44
+ Swin,g,0.2684375742673874,0.8763333333333333,0.9830383333333332,0.8885551216581556
45
+ Swin,h,0.25068944895267486,0.8873333333333333,0.9861691111111113,0.8974514563106796
46
+ Swin,i,0.08781581997871399,0.9746666666666667,0.9978362222222223,0.974950560316414
47
+ Swin,j,2.0428302302360533,0.42033333333333334,0.3025185555555555,0.11320754716981132
48
+ Swin,k,1.8622084745168686,0.5186666666666667,0.5179311666666666,0.1332533013205282
49
+ Swin,l,0.8495750360742048,0.7241816931944371,0.6664912098538803,0.5617543270038649
50
+ CaiT,a,0.40013275007637966,0.8390443256837472,0.9108057090239411,0.3694581280788177
51
+ CaiT,b,0.36606653323716165,0.8506758880855076,0.9150920810313077,0.3870967741935484
52
+ CaiT,c,0.45299853780328836,0.8003772398616787,0.8942338858195211,0.32085561497326204
53
+ CaiT,d,0.1995404364271233,0.9261238604212512,0.9594475138121547,0.5607476635514018
54
+ CaiT,e,0.4928394595302159,0.7771679473106476,0.8707636418678574,0.5964214711729622
55
+ CaiT,f,0.36103595997415405,0.850050344667338,0.9170773784465286,0.13416815742397137
56
+ CaiT,g,0.2526822371482849,0.9083333333333333,0.976861222222222,0.9132218365414957
57
+ CaiT,h,0.2987706809043884,0.8816666666666667,0.9732373333333334,0.8907356109572176
58
+ CaiT,i,0.16439563977718352,0.9483333333333334,0.9884212222222222,0.9491636602164644
59
+ CaiT,j,1.6938938024044037,0.48483333333333334,0.34579655555555555,0.18593626547274164
60
+ CaiT,k,1.6056072112321853,0.5248333333333334,0.5246175555555556,0.1984818667416362
61
+ CaiT,l,0.7488868967172212,0.7520490719686954,0.6676479234122561,0.5916572324305495
62
+ DeiT,a,0.4765459399334255,0.7749135491983653,0.850939226519337,0.28112449799196787
63
+ DeiT,b,0.29641423940058964,0.8814838101226029,0.9148747697974217,0.426179604261796
64
+ DeiT,c,0.4700488794977955,0.7830870795347376,0.8519852670349909,0.28865979381443296
65
+ DeiT,d,0.11101193318283434,0.9657340458975165,0.9766685082872928,0.7197943444730077
66
+ DeiT,e,0.4436056611305011,0.8035126234906695,0.8648906380080225,0.6100217864923747
67
+ DeiT,f,0.337409115804358,0.8522964913639532,0.8966829128564795,0.12802926383173296
68
+ DeiT,g,0.1696254106760025,0.9383333333333334,0.9919374444444445,0.9412884798476674
69
+ DeiT,h,0.26168069899082186,0.8861666666666667,0.9872772222222223,0.8967498110355253
70
+ DeiT,i,0.07133128488063813,0.983,0.998361111111111,0.9830957905203845
71
+ DeiT,j,1.940858449459076,0.4736666666666667,0.36180238888888894,0.10130904951622083
72
+ DeiT,k,1.8425643298625947,0.5183333333333333,0.5667426111111111,0.10967344423906346
73
+ DeiT,l,0.8096289309307544,0.7481360054994448,0.6678239118866791,0.5796487512134851
74
+ DeiT3,a,0.3153771912384543,0.8786545111600126,0.8877375690607736,0.396875
75
+ DeiT3,b,0.27676928310377796,0.8972021376925495,0.9046813996316758,0.43717728055077454
76
+ DeiT3,c,0.3605959014448869,0.8566488525620874,0.867316758747698,0.35774647887323946
77
+ DeiT3,d,0.23533754803987467,0.9091480666457089,0.9213038674033148,0.4677716390423573
78
+ DeiT3,e,0.42457847907305024,0.8111964873765093,0.853205176719897,0.596244131455399
79
+ DeiT3,f,0.28918993525220316,0.8904809852064132,0.8928482767899381,0.15227817745803357
80
+ DeiT3,g,0.17661127078533173,0.9428333333333333,0.9877949999999999,0.9447041753990005
81
+ DeiT3,h,0.2210533607006073,0.9213333333333333,0.9848492222222223,0.9254579911560329
82
+ DeiT3,i,0.1546455579996109,0.9491666666666667,0.9909291111111113,0.9505271695052717
83
+ DeiT3,j,1.796478444457054,0.5065,0.4410062222222223,0.17405857740585773
84
+ DeiT3,k,1.774512730240822,0.5128333333333334,0.45727222222222225,0.17592331547786863
85
+ DeiT3,l,0.7417711890545635,0.7793876579768388,0.6962303135333643,0.6175985334555454
86
+ Twins_SVT,a,0.4130411987707923,0.8233259981138007,0.891621546961326,0.3403755868544601
87
+ Twins_SVT,b,0.3827307312247214,0.8201823325998113,0.8948416206261509,0.33642691415313225
88
+ Twins_SVT,c,0.4311113485054039,0.8157812008802263,0.8846832412523021,0.3310502283105023
89
+ Twins_SVT,d,0.07931527397245478,0.9814523734674631,0.9895819521178638,0.830945558739255
90
+ Twins_SVT,e,0.5826191509594378,0.7189901207464325,0.8197532732914554,0.5311355311355311
91
+ Twins_SVT,f,0.3374182524886806,0.8535357447138099,0.9097097385998186,0.13296652911508483
92
+ Twins_SVT,g,0.22945928347110747,0.904,0.9866478888888889,0.9113300492610837
93
+ Twins_SVT,h,0.25510910618305205,0.9016666666666666,0.9884432222222223,0.9093701996927803
94
+ Twins_SVT,i,0.06859854513406753,0.9895,0.9990906666666667,0.9894701654688284
95
+ Twins_SVT,j,2.877535055875778,0.41883333333333334,0.1655298888888889,0.027336122733612273
96
+ Twins_SVT,k,2.7166742979884146,0.5043333333333333,0.4133546666666667,0.031901041666666664
97
+ Twins_SVT,l,1.0960884158819697,0.7418433715826768,0.5939750429978701,0.5637176050044683
98
+ Twins_PCPVT,a,0.44722607902332284,0.7843445457403332,0.8699972375690609,0.2883817427385892
99
+ Twins_PCPVT,b,0.34229608893356994,0.854133920150896,0.9080570902394106,0.3746630727762803
100
+ Twins_PCPVT,c,0.47921196653085174,0.7686262181703867,0.8568545119705341,0.27416173570019725
101
+ Twins_PCPVT,d,0.2515370008031815,0.913234831813895,0.9428913443830571,0.5018050541516246
102
+ Twins_PCPVT,e,0.48251375945287006,0.7661909989023051,0.858896541285098,0.5661914460285132
103
+ Twins_PCPVT,f,0.3832702880289431,0.8290604910541399,0.8924112354771649,0.11187122736418512
104
+ Twins_PCPVT,g,0.23884364533424376,0.9121666666666667,0.9764792222222223,0.9165743232547096
105
+ Twins_PCPVT,h,0.31143191051483154,0.8668333333333333,0.9660443333333333,0.8787372894217635
106
+ Twins_PCPVT,i,0.1907261505126953,0.9435,0.9881127777777777,0.9446891825746452
107
+ Twins_PCPVT,j,1.5205544946193694,0.48783333333333334,0.4457817777777778,0.1850967913020419
108
+ Twins_PCPVT,k,1.472436999797821,0.5191666666666667,0.4698568888888889,0.19480881942506278
109
+ Twins_PCPVT,l,0.7133773285155887,0.7375601501771456,0.6902435027241198,0.5768607724443686
110
+ PiT,a,0.4122267626378192,0.8280414963847846,0.8927302025782688,0.34803337306317045
111
+ PiT,b,0.39009387647344873,0.8230116315624018,0.8981012891344383,0.34152046783625734
112
+ PiT,c,0.4812444057415132,0.7843445457403332,0.8707458563535911,0.2985685071574642
113
+ PiT,d,0.08882246825332275,0.9710782772712984,0.9849594843462247,0.7604166666666666
114
+ PiT,e,0.498657209813006,0.7705817782656421,0.8615000378415197,0.5828343313373253
115
+ PiT,f,0.3495647843132805,0.848423824645651,0.9087592713952772,0.1298354824366385
116
+ PiT,g,0.21983683621883393,0.9066666666666666,0.9891565555555555,0.9137931034482759
117
+ PiT,h,0.2681618107557297,0.8861666666666667,0.988594888888889,0.8968122072820668
118
+ PiT,i,0.06011278629302978,0.9851666666666666,0.9989021111111112,0.9852282157676349
119
+ PiT,j,2.38970015335083,0.4266666666666667,0.21387577777777778,0.048672566371681415
120
+ PiT,k,2.2299761089086534,0.5051666666666667,0.5063448888888888,0.055961844197138316
121
+ PiT,l,0.9434709910593522,0.7408386653270583,0.6225121253803022,0.5664750110570544
122
+ Ensemble,a,,0.9107198994027036,0.9407062615101288,0.5086505190311419
123
+ Ensemble,b,,0.8880855077019805,0.9347661141804788,0.4523076923076923
124
+ Ensemble,c,,0.8792832442628105,0.928644567219153,0.4336283185840708
125
+ Ensemble,d,,0.9745363093366866,0.9857882136279925,0.784
126
+ Ensemble,e,,0.8386388583973655,0.901748278210853,0.6666666666666666
127
+ Ensemble,f,,0.9135620788474944,0.9448540229934943,0.20851063829787234
128
+ Ensemble,g,,0.9431666666666667,0.9957611111111111,0.9458987783595113
129
+ Ensemble,h,,0.9385,0.9963097777777778,0.9417153688200917
130
+ Ensemble,i,,0.989,0.9993681111111111,0.9890510948905109
131
+ Ensemble,j,,0.466,0.33056166666666664,0.0686046511627907
132
+ Ensemble,k,,0.5118333333333334,0.5398214444444445,0.074565560821485
133
+ Ensemble,l,,0.7875839458516207,0.6703077926895804,0.6177562089637454
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7966f3dfbded05c3a34f2cb24b867291d98289bbc3384a1ae7182983f24379d8
3
+ size 343287616
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40c54c3c6d7bc946e7b20ade559330bb0c0e87e1a25101c0a457ffd93f6ff9d2
3
+ size 343337390
roc_confusion_matrix/DeiT3_roc_confusion_matrix_a.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_b.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_c.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_d.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_e.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_f.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_g.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_h.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_i.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_j.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_k.png ADDED
roc_confusion_matrix/DeiT3_roc_confusion_matrix_l.png ADDED
roc_curves/DeiT3_ROC_a.png ADDED
roc_curves/DeiT3_ROC_b.png ADDED
roc_curves/DeiT3_ROC_c.png ADDED
roc_curves/DeiT3_ROC_d.png ADDED
roc_curves/DeiT3_ROC_e.png ADDED
roc_curves/DeiT3_ROC_f.png ADDED
roc_curves/DeiT3_ROC_g.png ADDED
roc_curves/DeiT3_ROC_h.png ADDED
roc_curves/DeiT3_ROC_i.png ADDED
roc_curves/DeiT3_ROC_j.png ADDED
roc_curves/DeiT3_ROC_k.png ADDED
roc_curves/DeiT3_ROC_l.png ADDED
training_curves/DeiT3_accuracy.png ADDED
training_curves/DeiT3_auc.png ADDED
training_curves/DeiT3_combined_metrics.png ADDED

Git LFS Details

  • SHA256: 7bcbc245acc7919848de82ce5734ae2e0a2a9d11dd72585f91fc3b2ed602bf74
  • Pointer size: 131 Bytes
  • Size of remote file: 143 kB
training_curves/DeiT3_f1.png ADDED
training_curves/DeiT3_loss.png ADDED
training_curves/DeiT3_metrics.csv ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1
2
+ 1,0.5185602677583695,0.6333154230117798,0.741575,0.629,0.8363617262499999,0.7078979999999999,0.7483690360272639,0.6630336058128974
3
+ 2,0.4191581697463989,0.5999163579940796,0.8087125,0.668,0.89944728875,0.7584039999999999,0.8113047016609329,0.7030411449016101
4
+ 3,0.3810452344179153,0.5638457331657409,0.8300875,0.705,0.9178470693750002,0.793352,0.832137520530521,0.7276084949215144
5
+ 4,0.3588511043548584,0.5365280117988587,0.8426,0.74,0.9274471615625001,0.816948,0.844015558803855,0.751434034416826
6
+ 5,0.34062150790691376,0.5135077581405639,0.8533625,0.761,0.93540629,0.835016,0.8544595114325769,0.7663734115347018
7
+ 6,0.3275874303817749,0.5005380935668945,0.8603625,0.773,0.9405587540625,0.847682,0.8613211177733915,0.7831900668576887
8
+ 7,0.3176137421131134,0.4901349854469299,0.8664625,0.776,0.9441472759375,0.8617779999999999,0.8674680858982471,0.7922077922077922
9
+ 8,0.3095633765697479,0.47415804529190064,0.8704875,0.791,0.947043950625,0.8690779999999999,0.8713382756522495,0.8018957345971564
10
+ 9,0.2999032303452492,0.4623113589286804,0.876125,0.803,0.95042684875,0.876306,0.8767842045058934,0.8118433619866284
11
+ 10,0.29264577033519745,0.45660440826416016,0.87925,0.805,0.95289195375,0.88329,0.8797939324556382,0.817244611059044
12
+ 11,0.28791155232191085,0.43909937000274657,0.8801875,0.824,0.954389264375,0.890128,0.8806900929833078,0.8301158301158301
13
+ 12,0.2837584602594376,0.43163455963134767,0.8825125,0.83,0.95554737125,0.8944559999999999,0.883113006926913,0.8352713178294574
14
+ 13,0.2774193372249603,0.426531772851944,0.8854375,0.832,0.9576246434375,0.897788,0.8858527107645938,0.8372093023255814
15
+ 14,0.27412921341657637,0.41923226332664487,0.88765,0.832,0.9585167934375001,0.9020180000000001,0.8881560936761156,0.8378378378378378
16
+ 15,0.26973266577720645,0.416291286945343,0.8894875,0.832,0.959919336875,0.90715,0.8898056860814398,0.8396946564885496
17
+ 16,0.26750720437765124,0.4073225519657135,0.8904875,0.843,0.9604387968750001,0.9082140000000001,0.8909034306705684,0.8477206595538312
18
+ 17,0.2649657944917679,0.39810547399520874,0.8925375,0.851,0.961098305625,0.911176,0.8929269781170991,0.8520357497517378
19
+ 18,0.25990938662290575,0.39820376777648925,0.8951,0.845,0.9627055256249999,0.913482,0.8954111518233256,0.8496605237633366
20
+ 19,0.2584483066916466,0.39332277154922485,0.8953875,0.85,0.9630241724999999,0.91674,0.8957562622223884,0.8552123552123552
21
+ 20,0.2571265205860138,0.3869056730270386,0.8956,0.857,0.9631486812499999,0.9183080000000001,0.8959952181709504,0.8599412340842312
22
+ 21,0.2537463493108749,0.38644637513160707,0.897625,0.851,0.9642951424999999,0.9205539999999999,0.8979261179520415,0.8563162970106075
23
+ 22,0.25238965204954145,0.3800084767341614,0.8975625,0.856,0.9645598528125001,0.922932,0.8977860929217337,0.8596491228070176
24
+ 23,0.24768273112773895,0.3703173868656158,0.9012625,0.862,0.9660763431250001,0.9243299999999999,0.9015615069227222,0.8614457831325302
25
+ 24,0.24780029238462448,0.37036190247535705,0.8994375,0.861,0.96568686875,0.925418,0.899657000311818,0.8625123639960435
26
+ 25,0.24634846239089966,0.3686402533054352,0.9003875,0.863,0.9661208728124999,0.9275059999999998,0.9006421046069447,0.8663414634146341
27
+ 26,0.2436532077908516,0.3603105943202972,0.900975,0.866,0.9668774356250001,0.928322,0.9011356545613378,0.8649193548387096
28
+ 27,0.24229492156505586,0.3620028879642487,0.90325,0.867,0.9672569853125,0.9305199999999999,0.9034768294507907,0.8702439024390244
29
+ 28,0.2393646076321602,0.36004478025436404,0.90375,0.868,0.96802906125,0.9315279999999999,0.9039085508910298,0.87109375
30
+ 29,0.23912578564882278,0.3569228014945984,0.9044625,0.872,0.9680420475,0.932168,0.9046948064093772,0.875
31
+ 30,0.23717505412101744,0.3505570204257965,0.9042625,0.873,0.9685092584375,0.93309,0.9044094704391997,0.8733798604187437
32
+ 31,0.23468575961589813,0.34703071355819703,0.90635,0.877,0.9693404349999999,0.9341380000000001,0.9065625701529022,0.8751269035532995
33
+ 32,0.23539003368616104,0.3494356880187988,0.90595,0.874,0.9688474803125,0.935418,0.906126013724267,0.876953125
34
+ 33,0.23233856045007706,0.34163496947288513,0.90735,0.879,0.9698404609375,0.9362,0.907481838380308,0.8774062816616008
35
+ 34,0.2332426217317581,0.34394970202445985,0.9068,0.878,0.9692798465625,0.936526,0.9068977573547775,0.8801571709233792
36
+ 35,0.2310953696370125,0.34073749589920044,0.9076,0.879,0.969895455625,0.937526,0.9078602946675641,0.8810226155358899
37
+ 36,0.23126315582990647,0.3418333249092102,0.9070375,0.879,0.969882454375,0.9380680000000001,0.9071315293265568,0.8819512195121951
38
+ 37,0.2311680959701538,0.33720108938217164,0.907275,0.882,0.9697331346875001,0.938762,0.9073745723347402,0.883629191321499
39
+ 38,0.22917891274094582,0.3327983841896057,0.9073125,0.883,0.9702823465625001,0.9404159999999999,0.9074363039434756,0.884272997032641
40
+ 39,0.22901247829198837,0.3326935448646545,0.908075,0.883,0.970331771875,0.941006,0.9081393024882582,0.8849557522123894
41
+ 40,0.22634558962583542,0.3289369740486145,0.909575,0.888,0.9712035128125001,0.9415620000000001,0.9096923998801558,0.8891089108910891
42
+ 41,0.2258337448835373,0.3322491147518158,0.9113,0.884,0.971314516875,0.9420200000000001,0.911470419442573,0.8869395711500975
43
+ 42,0.2235591392636299,0.3252280325889587,0.9112,0.887,0.9718439303125,0.942386,0.9114059810939565,0.8864321608040201
44
+ 43,0.22483974281549454,0.3271973593235016,0.9099875,0.887,0.9713250331250001,0.942542,0.9101458679078125,0.8891069676153092
45
+ 44,0.22383790327310563,0.32408277344703673,0.910975,0.891,0.9716544496874999,0.943662,0.9109994751443353,0.8923988153998026
46
+ 45,0.2236331503868103,0.324329913854599,0.9108125,0.888,0.9716207134374999,0.943964,0.9110182702500468,0.8904109589041096
47
+ 46,0.22264521746635438,0.3220202960968018,0.9115125,0.89,0.9718603571875,0.944584,0.911584337725598,0.8917322834645669
48
+ 47,0.22239046238660812,0.3180547444820404,0.9115625,0.896,0.9718298031249999,0.944894,0.9116055922737666,0.8953722334004024
49
+ 48,0.22044981991052628,0.31696475005149843,0.9132625,0.895,0.9724390790624999,0.9454100000000001,0.9133697049900747,0.8951048951048951
50
+ 49,0.21980776535272598,0.3161092948913574,0.9126375,0.894,0.9724316921875,0.9463440000000001,0.912773790951638,0.8954635108481263
51
+ 50,0.21873623955249785,0.3145533685684204,0.91375,0.894,0.972839791875,0.946132,0.9137931034482759,0.8946322067594433
52
+ 51,0.21824317857027054,0.31425856947898867,0.912525,0.895,0.97283375875,0.946282,0.9125599760095962,0.8959365708622399
53
+ 52,0.21919543809890746,0.3182931592464447,0.9134375,0.887,0.9725767028125001,0.9468200000000001,0.9135272155139044,0.8899707887049659
54
+ 53,0.21573848408460616,0.3131006507873535,0.9145,0.895,0.97356329,0.947044,0.9145235060358401,0.8959365708622399
55
+ 54,0.2184355439424515,0.3110634953975677,0.912775,0.898,0.9727923599999999,0.9476100000000001,0.9128708420737189,0.8988095238095238
56
+ 55,0.21798290065526962,0.30951609659194945,0.912525,0.897,0.97283212625,0.948216,0.9125992906029875,0.8979187314172448
57
+ 56,0.21562405709028243,0.30693556380271914,0.914725,0.902,0.9734204753125,0.9488059999999999,0.9148357135723559,0.9021956087824351
58
+ 57,0.21512275553941726,0.307545553445816,0.91455,0.895,0.9735012640625,0.9491660000000001,0.9145713571607098,0.896551724137931
59
+ 58,0.21433480132818222,0.3087418038845062,0.9153375,0.895,0.9738285909375,0.94894,0.9154020059704475,0.8967551622418879
60
+ 59,0.21392444179654121,0.30503026938438416,0.9157,0.899,0.9737282618750001,0.9495579999999998,0.9157252824152754,0.8997020854021847
61
+ 60,0.21512594406604768,0.3090649273395538,0.91385,0.889,0.9734550215625,0.94935,0.9138995077830248,0.8917073170731707
62
+ 61,0.2127853855729103,0.3083924367427826,0.91605,0.892,0.974092368125,0.949526,0.9160919540229885,0.8943248532289628
63
+ 62,0.21166097048521043,0.3073429772853851,0.9166125,0.891,0.974338465625,0.9500780000000001,0.9167093254092118,0.8938656280428432
64
+ 63,0.21225689854621888,0.3019155719280243,0.915525,0.9,0.974147106875,0.9505939999999999,0.9155967427686467,0.900990099009901
65
+ 64,0.2112337254822254,0.30575549268722535,0.9163125,0.891,0.9744269740625,0.951084,0.9164138482090465,0.8940719144800777
66
+ 65,0.21344662379026413,0.30367291092872617,0.914475,0.892,0.9736964775,0.950984,0.9144557525443224,0.8947368421052632
67
+ 66,0.21173548563718797,0.3027373013496399,0.9159625,0.893,0.9741931646875,0.951314,0.9160370171472818,0.895609756097561
68
+ 67,0.2099929867506027,0.2983927264213562,0.9167375,0.903,0.9746445490625001,0.9513940000000001,0.9167489470197848,0.9036742800397219
69
+ 68,0.20971158151030542,0.3016781272888184,0.9169125,0.891,0.974777525625,0.9515300000000001,0.9169176926442097,0.8938656280428432
70
+ 69,0.21095114467144013,0.2983397297859192,0.915675,0.903,0.9743319059375001,0.951198,0.9157318809803383,0.9036742800397219
71
+ 70,0.20855345443487167,0.29589474415779116,0.91805,0.903,0.9749649749999999,0.9515540000000001,0.9181911203174524,0.9029029029029029
72
+ 71,0.2086674389898777,0.2983514988422394,0.9175125,0.899,0.974977165,0.9520139999999999,0.9174949676806321,0.900883218842002
73
+ 72,0.20863597313165666,0.29659731197357175,0.9175625,0.9,0.974889679375,0.9522799999999999,0.9175779541336,0.9013806706114399
74
+ 73,0.2060047175168991,0.297880774974823,0.918525,0.897,0.9755669053125,0.9525119999999999,0.918496473765818,0.8991185112634672
75
+ 74,0.20731947827339173,0.2912002079486847,0.917325,0.903,0.9751874246875,0.953118,0.9173993405934658,0.9023162134944612
76
+ 75,0.2078201032757759,0.2942451949119568,0.917175,0.902,0.975040724375,0.9526840000000001,0.917187421886717,0.902970297029703
77
+ 76,0.20672744677066804,0.29390296936035154,0.9189625,0.901,0.97523655875,0.9532140000000001,0.9189716156932345,0.9024630541871921
78
+ 77,0.208072682929039,0.29404403591156003,0.9187,0.903,0.974856274375,0.9528859999999999,0.9187040647967601,0.904055390702275
79
+ 78,0.2095852622628212,0.2926036763191223,0.9172375,0.903,0.9744011346875,0.953614,0.9173067555921915,0.904055390702275
80
+ 79,0.20374466229081153,0.2943858309984207,0.9194,0.897,0.9761540275,0.9532560000000001,0.9195889659300643,0.8989205103042198
81
+ 80,0.20632467027902604,0.2900273082256317,0.9182875,0.904,0.9752186165625001,0.9537899999999998,0.9183048602172038,0.904572564612326
82
+ 81,0.20619098563194274,0.28777377593517306,0.918325,0.906,0.975357326875,0.9542839999999999,0.9183862103422433,0.905811623246493
83
+ 82,0.2064429716348648,0.28862507557868955,0.9185375,0.904,0.9752630987500001,0.953568,0.9185588782944477,0.903420523138833
84
+ 83,0.2061476641535759,0.28801716506481173,0.918725,0.903,0.9753162459375,0.954228,0.9186965437903266,0.9032901296111665
85
+ 84,0.2066561578154564,0.2883832335472107,0.9184875,0.905,0.9751403415625,0.9544360000000001,0.9185760485472049,0.9058473736372646
86
+ 85,0.2063444947838783,0.28657110333442687,0.9188125,0.908,0.9752569340625,0.9548200000000001,0.918841919804071,0.9085487077534792
87
+ 86,0.2049458182990551,0.2899364457130432,0.9191875,0.903,0.9755703328125,0.95466,0.9191763867531786,0.904621435594887
88
+ 87,0.20647942337989808,0.28670365738868714,0.9182625,0.905,0.9750774521875001,0.95514,0.9183777913696903,0.9060336300692384
89
+ 88,0.20370199925899504,0.2879558672904968,0.9188125,0.905,0.97592408875,0.9554059999999999,0.9188561148382745,0.9065880039331367
90
+ 89,0.2047832897424698,0.2840742385387421,0.9187625,0.908,0.9756098646875,0.9552479999999999,0.9187878787878788,0.9076305220883534
91
+ 90,0.20491388338804245,0.28839428901672365,0.9179125,0.905,0.9755635156249999,0.955358,0.9179340423138926,0.9065880039331367
92
+ 91,0.20328980213999748,0.2854208745956421,0.9189,0.908,0.9760068709375,0.9552739999999998,0.9189141900167471,0.9085487077534792
93
+ 92,0.20325400415062905,0.2835294172763824,0.920025,0.908,0.9760375984374999,0.9554100000000001,0.9201128758365797,0.9074446680080482
94
+ 93,0.20404619881510735,0.28677577114105224,0.9200625,0.906,0.9757791196875001,0.954948,0.9201313866789894,0.906930693069307
95
+ 94,0.20226166924834252,0.2831275086402893,0.91945,0.908,0.97624179125,0.9554060000000001,0.9195425260949908,0.907070707070707
96
+ 95,0.20287337729930877,0.28945638132095336,0.91985,0.897,0.9760771659375,0.9556419999999999,0.9198319621914931,0.8997078870496592
97
+ 96,0.2032047885775566,0.2804688057899475,0.9186125,0.909,0.9759240412499999,0.9561479999999999,0.9185890943646299,0.9083585095669687
98
+ 97,0.20307112154364587,0.28428847885131836,0.919525,0.907,0.975996040625,0.9555879999999999,0.9195511458775898,0.9080118694362018
99
+ 98,0.20383043421506883,0.2815167227983475,0.91935,0.908,0.975711323125,0.956012,0.9193096548274137,0.9076305220883534
100
+ 99,0.20145852105021478,0.2870875988006592,0.920475,0.898,0.9763022384375,0.9563520000000001,0.9204650581322665,0.9005847953216374
101
+ 100,0.20314387593269348,0.2811276526451111,0.919375,0.907,0.9758793290625001,0.9564219999999999,0.9194434730479093,0.9070929070929071
training_metrics.csv ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1
2
+ 1,0.5185602677583695,0.6333154230117798,0.741575,0.629,0.8363617262499999,0.7078979999999999,0.7483690360272639,0.6630336058128974
3
+ 2,0.4191581697463989,0.5999163579940796,0.8087125,0.668,0.89944728875,0.7584039999999999,0.8113047016609329,0.7030411449016101
4
+ 3,0.3810452344179153,0.5638457331657409,0.8300875,0.705,0.9178470693750002,0.793352,0.832137520530521,0.7276084949215144
5
+ 4,0.3588511043548584,0.5365280117988587,0.8426,0.74,0.9274471615625001,0.816948,0.844015558803855,0.751434034416826
6
+ 5,0.34062150790691376,0.5135077581405639,0.8533625,0.761,0.93540629,0.835016,0.8544595114325769,0.7663734115347018
7
+ 6,0.3275874303817749,0.5005380935668945,0.8603625,0.773,0.9405587540625,0.847682,0.8613211177733915,0.7831900668576887
8
+ 7,0.3176137421131134,0.4901349854469299,0.8664625,0.776,0.9441472759375,0.8617779999999999,0.8674680858982471,0.7922077922077922
9
+ 8,0.3095633765697479,0.47415804529190064,0.8704875,0.791,0.947043950625,0.8690779999999999,0.8713382756522495,0.8018957345971564
10
+ 9,0.2999032303452492,0.4623113589286804,0.876125,0.803,0.95042684875,0.876306,0.8767842045058934,0.8118433619866284
11
+ 10,0.29264577033519745,0.45660440826416016,0.87925,0.805,0.95289195375,0.88329,0.8797939324556382,0.817244611059044
12
+ 11,0.28791155232191085,0.43909937000274657,0.8801875,0.824,0.954389264375,0.890128,0.8806900929833078,0.8301158301158301
13
+ 12,0.2837584602594376,0.43163455963134767,0.8825125,0.83,0.95554737125,0.8944559999999999,0.883113006926913,0.8352713178294574
14
+ 13,0.2774193372249603,0.426531772851944,0.8854375,0.832,0.9576246434375,0.897788,0.8858527107645938,0.8372093023255814
15
+ 14,0.27412921341657637,0.41923226332664487,0.88765,0.832,0.9585167934375001,0.9020180000000001,0.8881560936761156,0.8378378378378378
16
+ 15,0.26973266577720645,0.416291286945343,0.8894875,0.832,0.959919336875,0.90715,0.8898056860814398,0.8396946564885496
17
+ 16,0.26750720437765124,0.4073225519657135,0.8904875,0.843,0.9604387968750001,0.9082140000000001,0.8909034306705684,0.8477206595538312
18
+ 17,0.2649657944917679,0.39810547399520874,0.8925375,0.851,0.961098305625,0.911176,0.8929269781170991,0.8520357497517378
19
+ 18,0.25990938662290575,0.39820376777648925,0.8951,0.845,0.9627055256249999,0.913482,0.8954111518233256,0.8496605237633366
20
+ 19,0.2584483066916466,0.39332277154922485,0.8953875,0.85,0.9630241724999999,0.91674,0.8957562622223884,0.8552123552123552
21
+ 20,0.2571265205860138,0.3869056730270386,0.8956,0.857,0.9631486812499999,0.9183080000000001,0.8959952181709504,0.8599412340842312
22
+ 21,0.2537463493108749,0.38644637513160707,0.897625,0.851,0.9642951424999999,0.9205539999999999,0.8979261179520415,0.8563162970106075
23
+ 22,0.25238965204954145,0.3800084767341614,0.8975625,0.856,0.9645598528125001,0.922932,0.8977860929217337,0.8596491228070176
24
+ 23,0.24768273112773895,0.3703173868656158,0.9012625,0.862,0.9660763431250001,0.9243299999999999,0.9015615069227222,0.8614457831325302
25
+ 24,0.24780029238462448,0.37036190247535705,0.8994375,0.861,0.96568686875,0.925418,0.899657000311818,0.8625123639960435
26
+ 25,0.24634846239089966,0.3686402533054352,0.9003875,0.863,0.9661208728124999,0.9275059999999998,0.9006421046069447,0.8663414634146341
27
+ 26,0.2436532077908516,0.3603105943202972,0.900975,0.866,0.9668774356250001,0.928322,0.9011356545613378,0.8649193548387096
28
+ 27,0.24229492156505586,0.3620028879642487,0.90325,0.867,0.9672569853125,0.9305199999999999,0.9034768294507907,0.8702439024390244
29
+ 28,0.2393646076321602,0.36004478025436404,0.90375,0.868,0.96802906125,0.9315279999999999,0.9039085508910298,0.87109375
30
+ 29,0.23912578564882278,0.3569228014945984,0.9044625,0.872,0.9680420475,0.932168,0.9046948064093772,0.875
31
+ 30,0.23717505412101744,0.3505570204257965,0.9042625,0.873,0.9685092584375,0.93309,0.9044094704391997,0.8733798604187437
32
+ 31,0.23468575961589813,0.34703071355819703,0.90635,0.877,0.9693404349999999,0.9341380000000001,0.9065625701529022,0.8751269035532995
33
+ 32,0.23539003368616104,0.3494356880187988,0.90595,0.874,0.9688474803125,0.935418,0.906126013724267,0.876953125
34
+ 33,0.23233856045007706,0.34163496947288513,0.90735,0.879,0.9698404609375,0.9362,0.907481838380308,0.8774062816616008
35
+ 34,0.2332426217317581,0.34394970202445985,0.9068,0.878,0.9692798465625,0.936526,0.9068977573547775,0.8801571709233792
36
+ 35,0.2310953696370125,0.34073749589920044,0.9076,0.879,0.969895455625,0.937526,0.9078602946675641,0.8810226155358899
37
+ 36,0.23126315582990647,0.3418333249092102,0.9070375,0.879,0.969882454375,0.9380680000000001,0.9071315293265568,0.8819512195121951
38
+ 37,0.2311680959701538,0.33720108938217164,0.907275,0.882,0.9697331346875001,0.938762,0.9073745723347402,0.883629191321499
39
+ 38,0.22917891274094582,0.3327983841896057,0.9073125,0.883,0.9702823465625001,0.9404159999999999,0.9074363039434756,0.884272997032641
40
+ 39,0.22901247829198837,0.3326935448646545,0.908075,0.883,0.970331771875,0.941006,0.9081393024882582,0.8849557522123894
41
+ 40,0.22634558962583542,0.3289369740486145,0.909575,0.888,0.9712035128125001,0.9415620000000001,0.9096923998801558,0.8891089108910891
42
+ 41,0.2258337448835373,0.3322491147518158,0.9113,0.884,0.971314516875,0.9420200000000001,0.911470419442573,0.8869395711500975
43
+ 42,0.2235591392636299,0.3252280325889587,0.9112,0.887,0.9718439303125,0.942386,0.9114059810939565,0.8864321608040201
44
+ 43,0.22483974281549454,0.3271973593235016,0.9099875,0.887,0.9713250331250001,0.942542,0.9101458679078125,0.8891069676153092
45
+ 44,0.22383790327310563,0.32408277344703673,0.910975,0.891,0.9716544496874999,0.943662,0.9109994751443353,0.8923988153998026
46
+ 45,0.2236331503868103,0.324329913854599,0.9108125,0.888,0.9716207134374999,0.943964,0.9110182702500468,0.8904109589041096
47
+ 46,0.22264521746635438,0.3220202960968018,0.9115125,0.89,0.9718603571875,0.944584,0.911584337725598,0.8917322834645669
48
+ 47,0.22239046238660812,0.3180547444820404,0.9115625,0.896,0.9718298031249999,0.944894,0.9116055922737666,0.8953722334004024
49
+ 48,0.22044981991052628,0.31696475005149843,0.9132625,0.895,0.9724390790624999,0.9454100000000001,0.9133697049900747,0.8951048951048951
50
+ 49,0.21980776535272598,0.3161092948913574,0.9126375,0.894,0.9724316921875,0.9463440000000001,0.912773790951638,0.8954635108481263
51
+ 50,0.21873623955249785,0.3145533685684204,0.91375,0.894,0.972839791875,0.946132,0.9137931034482759,0.8946322067594433
52
+ 51,0.21824317857027054,0.31425856947898867,0.912525,0.895,0.97283375875,0.946282,0.9125599760095962,0.8959365708622399
53
+ 52,0.21919543809890746,0.3182931592464447,0.9134375,0.887,0.9725767028125001,0.9468200000000001,0.9135272155139044,0.8899707887049659
54
+ 53,0.21573848408460616,0.3131006507873535,0.9145,0.895,0.97356329,0.947044,0.9145235060358401,0.8959365708622399
55
+ 54,0.2184355439424515,0.3110634953975677,0.912775,0.898,0.9727923599999999,0.9476100000000001,0.9128708420737189,0.8988095238095238
56
+ 55,0.21798290065526962,0.30951609659194945,0.912525,0.897,0.97283212625,0.948216,0.9125992906029875,0.8979187314172448
57
+ 56,0.21562405709028243,0.30693556380271914,0.914725,0.902,0.9734204753125,0.9488059999999999,0.9148357135723559,0.9021956087824351
58
+ 57,0.21512275553941726,0.307545553445816,0.91455,0.895,0.9735012640625,0.9491660000000001,0.9145713571607098,0.896551724137931
59
+ 58,0.21433480132818222,0.3087418038845062,0.9153375,0.895,0.9738285909375,0.94894,0.9154020059704475,0.8967551622418879
60
+ 59,0.21392444179654121,0.30503026938438416,0.9157,0.899,0.9737282618750001,0.9495579999999998,0.9157252824152754,0.8997020854021847
61
+ 60,0.21512594406604768,0.3090649273395538,0.91385,0.889,0.9734550215625,0.94935,0.9138995077830248,0.8917073170731707
62
+ 61,0.2127853855729103,0.3083924367427826,0.91605,0.892,0.974092368125,0.949526,0.9160919540229885,0.8943248532289628
63
+ 62,0.21166097048521043,0.3073429772853851,0.9166125,0.891,0.974338465625,0.9500780000000001,0.9167093254092118,0.8938656280428432
64
+ 63,0.21225689854621888,0.3019155719280243,0.915525,0.9,0.974147106875,0.9505939999999999,0.9155967427686467,0.900990099009901
65
+ 64,0.2112337254822254,0.30575549268722535,0.9163125,0.891,0.9744269740625,0.951084,0.9164138482090465,0.8940719144800777
66
+ 65,0.21344662379026413,0.30367291092872617,0.914475,0.892,0.9736964775,0.950984,0.9144557525443224,0.8947368421052632
67
+ 66,0.21173548563718797,0.3027373013496399,0.9159625,0.893,0.9741931646875,0.951314,0.9160370171472818,0.895609756097561
68
+ 67,0.2099929867506027,0.2983927264213562,0.9167375,0.903,0.9746445490625001,0.9513940000000001,0.9167489470197848,0.9036742800397219
69
+ 68,0.20971158151030542,0.3016781272888184,0.9169125,0.891,0.974777525625,0.9515300000000001,0.9169176926442097,0.8938656280428432
70
+ 69,0.21095114467144013,0.2983397297859192,0.915675,0.903,0.9743319059375001,0.951198,0.9157318809803383,0.9036742800397219
71
+ 70,0.20855345443487167,0.29589474415779116,0.91805,0.903,0.9749649749999999,0.9515540000000001,0.9181911203174524,0.9029029029029029
72
+ 71,0.2086674389898777,0.2983514988422394,0.9175125,0.899,0.974977165,0.9520139999999999,0.9174949676806321,0.900883218842002
73
+ 72,0.20863597313165666,0.29659731197357175,0.9175625,0.9,0.974889679375,0.9522799999999999,0.9175779541336,0.9013806706114399
74
+ 73,0.2060047175168991,0.297880774974823,0.918525,0.897,0.9755669053125,0.9525119999999999,0.918496473765818,0.8991185112634672
75
+ 74,0.20731947827339173,0.2912002079486847,0.917325,0.903,0.9751874246875,0.953118,0.9173993405934658,0.9023162134944612
76
+ 75,0.2078201032757759,0.2942451949119568,0.917175,0.902,0.975040724375,0.9526840000000001,0.917187421886717,0.902970297029703
77
+ 76,0.20672744677066804,0.29390296936035154,0.9189625,0.901,0.97523655875,0.9532140000000001,0.9189716156932345,0.9024630541871921
78
+ 77,0.208072682929039,0.29404403591156003,0.9187,0.903,0.974856274375,0.9528859999999999,0.9187040647967601,0.904055390702275
79
+ 78,0.2095852622628212,0.2926036763191223,0.9172375,0.903,0.9744011346875,0.953614,0.9173067555921915,0.904055390702275
80
+ 79,0.20374466229081153,0.2943858309984207,0.9194,0.897,0.9761540275,0.9532560000000001,0.9195889659300643,0.8989205103042198
81
+ 80,0.20632467027902604,0.2900273082256317,0.9182875,0.904,0.9752186165625001,0.9537899999999998,0.9183048602172038,0.904572564612326
82
+ 81,0.20619098563194274,0.28777377593517306,0.918325,0.906,0.975357326875,0.9542839999999999,0.9183862103422433,0.905811623246493
83
+ 82,0.2064429716348648,0.28862507557868955,0.9185375,0.904,0.9752630987500001,0.953568,0.9185588782944477,0.903420523138833
84
+ 83,0.2061476641535759,0.28801716506481173,0.918725,0.903,0.9753162459375,0.954228,0.9186965437903266,0.9032901296111665
85
+ 84,0.2066561578154564,0.2883832335472107,0.9184875,0.905,0.9751403415625,0.9544360000000001,0.9185760485472049,0.9058473736372646
86
+ 85,0.2063444947838783,0.28657110333442687,0.9188125,0.908,0.9752569340625,0.9548200000000001,0.918841919804071,0.9085487077534792
87
+ 86,0.2049458182990551,0.2899364457130432,0.9191875,0.903,0.9755703328125,0.95466,0.9191763867531786,0.904621435594887
88
+ 87,0.20647942337989808,0.28670365738868714,0.9182625,0.905,0.9750774521875001,0.95514,0.9183777913696903,0.9060336300692384
89
+ 88,0.20370199925899504,0.2879558672904968,0.9188125,0.905,0.97592408875,0.9554059999999999,0.9188561148382745,0.9065880039331367
90
+ 89,0.2047832897424698,0.2840742385387421,0.9187625,0.908,0.9756098646875,0.9552479999999999,0.9187878787878788,0.9076305220883534
91
+ 90,0.20491388338804245,0.28839428901672365,0.9179125,0.905,0.9755635156249999,0.955358,0.9179340423138926,0.9065880039331367
92
+ 91,0.20328980213999748,0.2854208745956421,0.9189,0.908,0.9760068709375,0.9552739999999998,0.9189141900167471,0.9085487077534792
93
+ 92,0.20325400415062905,0.2835294172763824,0.920025,0.908,0.9760375984374999,0.9554100000000001,0.9201128758365797,0.9074446680080482
94
+ 93,0.20404619881510735,0.28677577114105224,0.9200625,0.906,0.9757791196875001,0.954948,0.9201313866789894,0.906930693069307
95
+ 94,0.20226166924834252,0.2831275086402893,0.91945,0.908,0.97624179125,0.9554060000000001,0.9195425260949908,0.907070707070707
96
+ 95,0.20287337729930877,0.28945638132095336,0.91985,0.897,0.9760771659375,0.9556419999999999,0.9198319621914931,0.8997078870496592
97
+ 96,0.2032047885775566,0.2804688057899475,0.9186125,0.909,0.9759240412499999,0.9561479999999999,0.9185890943646299,0.9083585095669687
98
+ 97,0.20307112154364587,0.28428847885131836,0.919525,0.907,0.975996040625,0.9555879999999999,0.9195511458775898,0.9080118694362018
99
+ 98,0.20383043421506883,0.2815167227983475,0.91935,0.908,0.975711323125,0.956012,0.9193096548274137,0.9076305220883534
100
+ 99,0.20145852105021478,0.2870875988006592,0.920475,0.898,0.9763022384375,0.9563520000000001,0.9204650581322665,0.9005847953216374
101
+ 100,0.20314387593269348,0.2811276526451111,0.919375,0.907,0.9758793290625001,0.9564219999999999,0.9194434730479093,0.9070929070929071