Upload PiT model from experiment b3

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+training_curves/PiT_combined_metrics.png filter=lfs diff=lfs merge=lfs -text
+training_notebook_b3.ipynb filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,165 @@
+---
+license: apache-2.0
+tags:
+- vision-transformer
+- image-classification
+- pytorch
+- timm
+- pit
+- gravitational-lensing
+- strong-lensing
+- astronomy
+- astrophysics
+datasets:
+- parlange/gravit-j24
+metrics:
+- accuracy
+- auc
+- f1
+paper:
+- title: "GraViT: A Gravitational Lens Discovery Toolkit with Vision Transformers"
+  url: "https://arxiv.org/abs/2509.00226"
+  authors: "Parlange et al."
+model-index:
+- name: PiT-b3
+  results:
+  - task:
+      type: image-classification
+      name: Strong Gravitational Lens Discovery
+    dataset:
+      type: common-test-sample
+      name: Common Test Sample (More et al. 2024)
+    metrics:
+    - type: accuracy
+      value: 0.8191
+      name: Average Accuracy
+    - type: auc
+      value: 0.7349
+      name: Average AUC-ROC
+    - type: f1
+      value: 0.4719
+      name: Average F1-Score
+---
+
+# 🌌 pit-gravit-b3
+
+🔭 This model is part of **GraViT**: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery
+
+🔗 **GitHub Repository**: [https://github.com/parlange/gravit](https://github.com/parlange/gravit)
+
+## 🛰️ Model Details
+
+- **🤖 Model Type**: PiT
+- **🧪 Experiment**: B3 - J24-all-blocks
+- **🌌 Dataset**: J24
+- **🪐 Fine-tuning Strategy**: all-blocks
+
+
+
+## 💻 Quick Start
+
+```python
+import torch
+import timm
+
+# Load the model directly from the Hub
+model = timm.create_model(
+    'hf-hub:parlange/pit-gravit-b3',
+    pretrained=True
+)
+model.eval()
+
+# Example inference
+dummy_input = torch.randn(1, 3, 224, 224)
+with torch.no_grad():
+    output = model(dummy_input)
+    predictions = torch.softmax(output, dim=1)
+print(f"Lens probability: {predictions[0][1]:.4f}")
+```
+
+## ⚡️ Training Configuration
+
+**Training Dataset:** J24 (Jaelani et al. 2024)  
+**Fine-tuning Strategy:** all-blocks
+
+
+| 🔧 Parameter | 📝 Value |
+|--------------|----------|
+| Batch Size | 192 |
+| Learning Rate | AdamW with ReduceLROnPlateau |
+| Epochs | 100 |
+| Patience | 10 |
+| Optimizer | AdamW |
+| Scheduler | ReduceLROnPlateau |
+| Image Size | 224x224 |
+| Fine Tune Mode | all_blocks |
+| Stochastic Depth Probability | 0.1 |
+
+
+## 📈 Training Curves
+
+![Combined Training Metrics](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/training_curves/PiT_combined_metrics.png)
+
+
+## 🏁 Final Epoch Training Metrics
+
+| Metric | Training | Validation |
+|:---------:|:-----------:|:-------------:|
+| 📉 Loss | 0.0103 | 0.0629 |
+| 🎯 Accuracy | 0.9964 | 0.9871 |
+| 📊 AUC-ROC | 0.9999 | 0.9983 |
+| ⚖️ F1 Score | 0.9964 | 0.9871 |
+
+
+## ☑️ Evaluation Results
+
+### ROC Curves and Confusion Matrices
+
+Performance across all test datasets (a through l) in the Common Test Sample (More et al. 2024):
+
+![ROC + Confusion Matrix - Dataset A](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_a.png)
+![ROC + Confusion Matrix - Dataset B](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_b.png)
+![ROC + Confusion Matrix - Dataset C](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_c.png)
+![ROC + Confusion Matrix - Dataset D](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_d.png)
+![ROC + Confusion Matrix - Dataset E](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_e.png)
+![ROC + Confusion Matrix - Dataset F](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_f.png)
+![ROC + Confusion Matrix - Dataset G](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_g.png)
+![ROC + Confusion Matrix - Dataset H](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_h.png)
+![ROC + Confusion Matrix - Dataset I](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_i.png)
+![ROC + Confusion Matrix - Dataset J](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_j.png)
+![ROC + Confusion Matrix - Dataset K](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_k.png)
+![ROC + Confusion Matrix - Dataset L](https://huggingface.co/parlange/pit-gravit-b3/resolve/main/roc_confusion_matrix/PiT_roc_confusion_matrix_l.png)
+
+### 📋 Performance Summary
+
+Average performance across 12 test datasets from the Common Test Sample (More et al. 2024):
+
+| Metric | Value |
+|-----------|----------|
+| 🎯 Average Accuracy | 0.8191 |
+| 📈 Average AUC-ROC | 0.7349 |
+| ⚖️ Average F1-Score | 0.4719 |
+
+
+## 📘 Citation
+
+If you use this model in your research, please cite:
+
+```bibtex
+@misc{parlange2025gravit,
+      title={GraViT: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery}, 
+      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},
+      year={2025},
+      eprint={2509.00226},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2509.00226}, 
+}
+```
+
+---
+
+
+## Model Card Contact
+
+For questions about this model, please contact the author through: https://github.com/parlange/

config.json

@@ -0,0 +1,76 @@
+{
+  "architecture": "vit_base_patch16_224",
+  "num_classes": 2,
+  "num_features": 1000,
+  "global_pool": "avg",
+  "crop_pct": 0.875,
+  "interpolation": "bicubic",
+  "mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "first_conv": "conv1",
+  "classifier": "fc",
+  "input_size": [
+    3,
+    224,
+    224
+  ],
+  "pool_size": [
+    7,
+    7
+  ],
+  "pretrained_cfg": {
+    "tag": "gravit_b3",
+    "custom_load": false,
+    "input_size": [
+      3,
+      224,
+      224
+    ],
+    "fixed_input_size": true,
+    "interpolation": "bicubic",
+    "crop_pct": 0.875,
+    "crop_mode": "center",
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ],
+    "num_classes": 2,
+    "pool_size": [
+      7,
+      7
+    ],
+    "first_conv": "conv1",
+    "classifier": "fc"
+  },
+  "model_name": "pit_gravit_b3",
+  "experiment": "b3",
+  "training_strategy": "all-blocks",
+  "dataset": "J24",
+  "hyperparameters": {
+    "batch_size": "192",
+    "learning_rate": "AdamW with ReduceLROnPlateau",
+    "epochs": "100",
+    "patience": "10",
+    "optimizer": "AdamW",
+    "scheduler": "ReduceLROnPlateau",
+    "image_size": "224x224",
+    "fine_tune_mode": "all_blocks",
+    "stochastic_depth_probability": "0.1"
+  },
+  "hf_hub_id": "parlange/pit-gravit-b3",
+  "license": "apache-2.0"
+}

evaluation_results.csv

@@ -0,0 +1,133 @@
+Model,Dataset,Loss,Accuracy,AUCROC,F1
+ViT,a,0.3790108698956886,0.9188934297390757,0.7895534069981583,0.40825688073394495
+ViT,b,0.6474953413815365,0.8736246463376297,0.7928941068139963,0.30689655172413793
+ViT,c,0.3887653840258975,0.9154353976736875,0.7815948434622468,0.3982102908277405
+ViT,d,0.4339898591969456,0.9006601697579377,0.7802946593001843,0.3603238866396761
+ViT,e,0.7275202240136531,0.8770581778265643,0.8333762203890108,0.6137931034482759
+ViT,f,0.32068420921553115,0.923398652311982,0.7887962050752345,0.15252784918594686
+ViT,g,2.4499126282334327,0.5935,0.7337697777777779,0.41664673523080603
+ViT,h,2.3127426176071166,0.6156666666666667,0.7213267777777778,0.43033596837944665
+ViT,i,2.3367191193699837,0.6078333333333333,0.7155961666666667,0.4253968253968254
+ViT,j,0.43574089199304583,0.9035,0.9639603333333334,0.9041549412348949
+ViT,k,0.3225473812222481,0.9178333333333333,0.9721483333333332,0.9172124265323258
+ViT,l,0.9802426720999076,0.8208978901168632,0.848824705820602,0.685486117559662
+MLP-Mixer,a,0.24821871188368222,0.9342973907576234,0.7274953959484346,0.40114613180515757
+MLP-Mixer,b,0.3468351167483511,0.8972021376925495,0.7498066298342543,0.29978586723768735
+MLP-Mixer,c,0.22863765820715645,0.9415278214397989,0.7362136279926336,0.4294478527607362
+MLP-Mixer,d,0.22197297275188171,0.9402703552342031,0.7448802946593002,0.42424242424242425
+MLP-Mixer,e,0.5210685612128804,0.8682766190998902,0.8295996367214108,0.5384615384615384
+MLP-Mixer,f,0.1573308274558462,0.9544574393927658,0.7447600612812645,0.19230769230769232
+MLP-Mixer,g,2.0364777975082395,0.5281666666666667,0.5996590555555555,0.21382949180783115
+MLP-Mixer,h,1.9738134279251098,0.5516666666666666,0.5622524444444443,0.22254335260115607
+MLP-Mixer,i,1.9702800275236367,0.551,0.5762447222222222,0.2222863741339492
+MLP-Mixer,j,0.35843076133728025,0.875,0.9505481111111111,0.8680042238648363
+MLP-Mixer,k,0.2922329999357462,0.8978333333333334,0.9618148888888888,0.8894499549143372
+MLP-Mixer,l,0.7973672104744812,0.8023901433028396,0.7688593430466856,0.6098757699133521
+CvT,a,0.5576080598665536,0.8827412763281987,0.7585920810313076,0.33511586452762926
+CvT,b,0.8406257543388458,0.839987425337944,0.7651546961325967,0.2697274031563845
+CvT,c,0.45941355157408165,0.9053756680289218,0.7661510128913444,0.38445807770961143
+CvT,d,0.4890136360203984,0.899402703552342,0.8040036832412524,0.3700787401574803
+CvT,e,0.7525440884670233,0.8616904500548848,0.7940134715810185,0.5987261146496815
+CvT,f,0.4638477216021927,0.900782278677097,0.7746531228706713,0.1279782164737917
+CvT,g,2.635408263206482,0.5545,0.6485876666666667,0.359146487652841
+CvT,h,2.433302253484726,0.5891666666666666,0.6446558888888889,0.3779964673227353
+CvT,i,2.4489952618479727,0.586,0.699406611111111,0.3761928679055751
+CvT,j,0.6779288778305054,0.8571666666666666,0.929701777777778,0.8568565224653416
+CvT,k,0.4915158650279045,0.8886666666666667,0.9551427222222222,0.8847878578820283
+CvT,l,1.142270628382002,0.7902279096821956,0.8003466836323919,0.6321060929240471
+Swin,a,0.4143876506730619,0.8641936497956617,0.7043434622467771,0.2627986348122867
+Swin,b,0.6756683163993653,0.8022634391700723,0.720195211786372,0.1966794380587484
+Swin,c,0.3583531161967556,0.8934297390757623,0.7160773480662983,0.31237322515212984
+Swin,d,0.29349016994284743,0.9135491983652939,0.7763278084714549,0.358974358974359
+Swin,e,0.5968914167254215,0.845225027442371,0.7776356618481798,0.5220338983050847
+Swin,f,0.3336647423351405,0.8915653318875377,0.7320114316466519,0.0990990990990991
+Swin,g,2.182571418762207,0.49166666666666664,0.5955143333333334,0.2375
+Swin,h,2.0143414651155473,0.54,0.5570215555555555,0.2560646900269542
+Swin,i,1.9799532911777495,0.5506666666666666,0.6583314444444445,0.2605595172792101
+Swin,j,0.3886156997680664,0.8788333333333334,0.964886111111111,0.8850229321524593
+Swin,k,0.1859975779056549,0.9378333333333333,0.9823477777777778,0.9375104707656223
+Swin,l,0.8632472551865445,0.7817672254243562,0.7818820389829563,0.618823312090145
+CaiT,a,0.23817305107586642,0.9374410562716127,0.8123416206261509,0.5134474327628362
+CaiT,b,0.3217430385037512,0.9104055328513047,0.8156022099447514,0.42424242424242425
+CaiT,c,0.23880257207580574,0.9374410562716127,0.8067292817679558,0.5134474327628362
+CaiT,d,0.28198474085357195,0.9173215969820812,0.8376279926335175,0.4439746300211416
+CaiT,e,0.5301304184284744,0.8682766190998902,0.8315673957466131,0.6363636363636364
+CaiT,f,0.19037221234553495,0.9409805592130741,0.8188489798752676,0.21604938271604937
+CaiT,g,1.6843535647392274,0.6265,0.7207983333333333,0.46349054345223845
+CaiT,h,1.6403812870979309,0.6408333333333334,0.7088211111111111,0.4732339281349303
+CaiT,i,1.663275026500225,0.6301666666666667,0.7565758888888889,0.46594464500601684
+CaiT,j,0.23444856452941895,0.9306666666666666,0.9765742777777777,0.9306897700766411
+CaiT,k,0.21337003868818283,0.9343333333333333,0.9810072777777779,0.934113712374582
+CaiT,l,0.6691959589620489,0.8413092908889006,0.851808223589687,0.7203950433243268
+DeiT,a,0.27532369791711886,0.9355548569632192,0.7400883977900553,0.4383561643835616
+DeiT,b,0.37732297870225406,0.9009745363093367,0.7632817679558012,0.3368421052631579
+DeiT,c,0.27752348167909013,0.9393272555800063,0.7630893186003682,0.45325779036827196
+DeiT,d,0.27014286825489003,0.9364979566174159,0.795318600368324,0.4419889502762431
+DeiT,e,0.6625746366344024,0.8726673984632273,0.80245213047756,0.5797101449275363
+DeiT,f,0.17108403316157025,0.9514367593524902,0.767566717155716,0.20330368487928843
+DeiT,g,2.1656926336288453,0.59,0.6878379444444445,0.3800403225806452
+DeiT,h,2.11278225171566,0.6103333333333333,0.6916527777777778,0.3920956838273531
+DeiT,i,2.108869301110506,0.6088333333333333,0.7280663333333334,0.39118028534370947
+DeiT,j,0.31411294651031496,0.9131666666666667,0.9682927222222223,0.9117995598442525
+DeiT,k,0.25728961780667303,0.932,0.9767967777777777,0.9295823265447014
+DeiT,l,0.83515618283903,0.8318439003754429,0.831689744806227,0.6892710572601134
+DeiT3,a,0.27852537169354546,0.9038038352719271,0.7915755064456723,0.39763779527559057
+DeiT3,b,0.33796260499590713,0.8667085822068532,0.8198545119705342,0.3226837060702875
+DeiT3,c,0.26950191666367146,0.9075762338887142,0.800132596685083,0.40725806451612906
+DeiT3,d,0.2598652228155589,0.908519333542911,0.8328038674033148,0.40973630831643004
+DeiT3,e,0.3859063482278003,0.8792535675082327,0.8585105577840006,0.6474358974358975
+DeiT3,f,0.2226914082389882,0.9144140655255208,0.8138108526862634,0.1545524100994644
+DeiT3,g,1.2969025439023971,0.6038333333333333,0.732013,0.4486198097889121
+DeiT3,h,1.2606069605350494,0.6255,0.7032497222222223,0.4625687634537192
+DeiT3,i,1.2554979050159454,0.626,0.750082888888889,0.46290090952608903
+DeiT3,j,0.2535483182668686,0.9056666666666666,0.9677222222222222,0.907546553413917
+DeiT3,k,0.2121436755657196,0.9278333333333333,0.9726957777777777,0.9277007847720822
+DeiT3,l,0.559908740118049,0.8223256305853736,0.8453933340814723,0.6959826275787188
+Twins_SVT,a,0.32229545287125066,0.9047469349261239,0.7872716390423573,0.40471512770137524
+Twins_SVT,b,0.4744058540550578,0.858220685319082,0.799073664825046,0.3135464231354642
+Twins_SVT,c,0.23853482503686377,0.9327255580006287,0.8031399631675874,0.49047619047619045
+Twins_SVT,d,0.20696176561834076,0.9386985224772084,0.8572504604051567,0.513715710723192
+Twins_SVT,e,0.4441066685037608,0.8792535675082327,0.828918489366533,0.6518987341772152
+Twins_SVT,f,0.24040106920338317,0.9255673456742313,0.8126722450556175,0.17652099400171378
+Twins_SVT,g,1.6869367780685425,0.5768333333333333,0.6634996111111111,0.3964820537199905
+Twins_SVT,h,1.5618858572244645,0.6163333333333333,0.6534029444444445,0.42015113350125943
+Twins_SVT,i,1.5451468470990657,0.6195,0.7416233888888889,0.42217160212604404
+Twins_SVT,j,0.4541445074081421,0.8555,0.9362851666666666,0.8525259397856778
+Twins_SVT,k,0.3123545649945736,0.8981666666666667,0.9654593888888889,0.8913391428063311
+Twins_SVT,l,0.7169495407413323,0.8085241393897732,0.811499193039308,0.6553726087370324
+Twins_PCPVT,a,0.3407551253865028,0.9157497642250865,0.7543489871086556,0.36792452830188677
+Twins_PCPVT,b,0.4568046205300874,0.8827412763281987,0.7560791896869244,0.2948960302457467
+Twins_PCPVT,c,0.3403567170729543,0.9163784973278843,0.7330865561694292,0.3696682464454976
+Twins_PCPVT,d,0.2872482358388127,0.9276956931782459,0.7866114180478823,0.40414507772020725
+Twins_PCPVT,e,0.6489756450539559,0.8704720087815587,0.8124347233784909,0.5693430656934306
+Twins_PCPVT,f,0.22982485620439833,0.9347068391294245,0.7606799529540434,0.15615615615615616
+Twins_PCPVT,g,2.0454217346906662,0.5676666666666667,0.6946129999999999,0.3426254434870755
+Twins_PCPVT,h,1.9836849460601806,0.5855,0.667161,0.3521750455847877
+Twins_PCPVT,i,1.9555286951363087,0.5915,0.7357337222222222,0.35550880883513014
+Twins_PCPVT,j,0.3498825296163559,0.9013333333333333,0.9614410555555556,0.9004707464694015
+Twins_PCPVT,k,0.2599894929230213,0.9251666666666667,0.9737956666666667,0.9226528854435831
+Twins_PCPVT,l,0.8202423188098834,0.8155042039024906,0.8326626157015399,0.662996232975949
+PiT,a,0.43440851872726566,0.9195221628418736,0.6457163904235728,0.37254901960784315
+PiT,b,0.6821634257150798,0.8792832442628105,0.7135303867403315,0.2835820895522388
+PiT,c,0.4489740155864459,0.9166928638792833,0.6253001841620626,0.3645083932853717
+PiT,d,0.36322150241931406,0.9352404904118202,0.7464861878453039,0.4245810055865922
+PiT,e,0.8982287239688157,0.8726673984632273,0.8314765760992961,0.5671641791044776
+PiT,f,0.29423939931023113,0.9374951591666021,0.6912865159517909,0.15849843587069865
+PiT,g,3.0194950482845306,0.562,0.6416187777777778,0.33129770992366414
+PiT,h,2.895865800142288,0.5818333333333333,0.522822,0.3416426134872737
+PiT,i,2.8504026728868483,0.5916666666666667,0.6812351666666667,0.34701492537313433
+PiT,j,0.4774704883098602,0.8945,0.9592785555555556,0.8931645569620253
+PiT,k,0.30837811005115506,0.9241666666666667,0.975773277777778,0.9208282582216809
+PiT,l,1.1643054382698763,0.8143937390936492,0.7846461923134944,0.6577613104524181
+Ensemble,a,,0.9289531593838416,0.7939346224677717,0.44607843137254904
+Ensemble,b,,0.8783401446086136,0.8044364640883978,0.31985940246045697
+Ensemble,c,,0.9383841559258095,0.7978895027624309,0.48148148148148145
+Ensemble,d,,0.9383841559258095,0.8286721915285451,0.48148148148148145
+Ensemble,e,,0.8880351262349067,0.850836297585711,0.6408450704225352
+Ensemble,f,,0.9421423592285648,0.8087909536354286,0.19590958019375673
+Ensemble,g,,0.5633333333333334,0.7028778888888889,0.34071464519375944
+Ensemble,h,,0.5951666666666666,0.6861525555555557,0.35791699709225483
+Ensemble,i,,0.5951666666666666,0.73121,0.35791699709225483
+Ensemble,j,,0.9088333333333334,0.9700168888888888,0.9095419216140235
+Ensemble,k,,0.9406666666666667,0.9819625555555556,0.9392076502732241
+Ensemble,l,,0.8244408016498335,0.8396341104616648,0.6794129007338741

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b21dce370730d57725c0300c4ef486825afacdbbf82ccc501d031fd1139d03f9
+size 290985688

pit-gravit-b3.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01e02d406a585b90d523b4f5804cb44e7f626aedea9d00507365e2e56c3d1129
+size 291039882

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01e02d406a585b90d523b4f5804cb44e7f626aedea9d00507365e2e56c3d1129
+size 291039882

training_curves/PiT_metrics.csv

@@ -0,0 +1,36 @@
+epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1
+1,0.14566365511793838,0.08487027064205366,0.9427709144243245,0.9694533762057878,0.9851093454718796,0.994546168877493,0.9418307961354319,0.9689034369885434
+2,0.07756046253042306,0.07278770848006681,0.9740508828340949,0.977491961414791,0.9945806704518922,0.9966455635855249,0.9738085748448443,0.9775641025641025
+3,0.06155229155601513,0.06513803804419048,0.9785919783381283,0.9796355841371919,0.9965701100870393,0.9956719728796114,0.9784332111499446,0.9794372294372294
+4,0.05817294309795518,0.05396232030495187,0.9809048344333503,0.9828510182207931,0.9966998044392082,0.9976949737446423,0.980759939750476,0.9826839826839827
+5,0.05050496557747315,0.05946147611957654,0.9830484571557511,0.9828510182207931,0.9974789077145225,0.9974652178258427,0.9829392227552729,0.9827956989247312
+6,0.04554648937070848,0.07530270049713815,0.9841766796412252,0.9839228295819936,0.9980721664550192,0.997233164347855,0.9840828486310115,0.9836779107725789
+7,0.045750615091004296,0.07067370211100656,0.9848536131325097,0.9823151125401929,0.9979605836791916,0.9973560837644129,0.9847620669107006,0.9823245848955544
+8,0.041395390699091934,0.05189927040907731,0.9865741524228578,0.9839228295819936,0.9983463015433974,0.9980631576045188,0.9865102306863912,0.9838536060279871
+9,0.038652644766771084,0.06319661472004709,0.9872510859141422,0.9812433011789925,0.998584051186131,0.9982010111557986,0.9871918390478889,0.9809471965160588
+10,0.03613892484208536,0.059335076348957905,0.9882946917132058,0.9823151125401929,0.9987184138476197,0.9980861331963987,0.9882426268521404,0.9824374667376264
+11,0.03727046646559817,0.052120874501117936,0.9881254583403847,0.9844587352625938,0.9986679980162962,0.9983072732682435,0.98807264073434,0.984366576819407
+12,0.033966494260936154,0.04754341436831514,0.9886049528967112,0.9844587352625938,0.9988358900855194,0.9984238743970343,0.9885578339186587,0.9843497031840259
+13,0.031665333392081836,0.06848823487566982,0.9895639420093643,0.9817792068595927,0.9991228049491042,0.9981137039066548,0.9895255350469936,0.9817792068595927
+14,0.034051497119330254,0.07028256035694355,0.9884921306481638,0.9817792068595927,0.9989677658460311,0.9976398323241303,0.9884484711211778,0.9815016322089227
+15,0.03127622385937226,0.04463923775210641,0.9896485586957748,0.9855305466237942,0.9990817177808828,0.9985140535951632,0.9896089923270761,0.9854604200323102
+16,0.029261085257816156,0.06298187336737704,0.9902690810627855,0.9844587352625938,0.999124038057394,0.9980832612474138,0.9902379672335248,0.9843158464034614
+17,0.0289509687084124,0.07007141667642777,0.990184464376375,0.9844587352625938,0.9992388682835698,0.9978041078060722,0.9901561439239647,0.984400215169446
+18,0.029313751012666877,0.048654140910535,0.990381903311333,0.9871382636655949,0.9992080039808491,0.9981320843801588,0.9903473264075636,0.9870828848223897
+19,0.027540785092642957,0.05571309371680692,0.9907203700569752,0.9823151125401929,0.9993202518395994,0.9980108881329919,0.990686482661005,0.9822294022617124
+20,0.026049582501074288,0.051708202365893643,0.9913126868618491,0.9860664523043944,0.999335620346661,0.9976949737446423,0.9912861427035591,0.986021505376344
+21,0.024673762814315024,0.06106355182610907,0.991538331358944,0.9855305466237942,0.9994650585305271,0.9986467376382699,0.9915139171758316,0.9853975121687398
+22,0.016652436815372457,0.05243686088315927,0.9946409431939979,0.9876741693461951,0.9996888462052035,0.9988460508978287,0.9946318585070916,0.987601078167116
+23,0.015147703151754628,0.049660267170602485,0.9949512043775033,0.9882100750267953,0.9997919563337484,0.9988615594223478,0.994941073396829,0.9881465517241379
+24,0.013805163181409742,0.057504319061803665,0.9953178766852824,0.984994640943194,0.9998115826444022,0.9988247984753398,0.9953099395377748,0.9848648648648649
+25,0.012066704250700654,0.05450154134219099,0.9959101934901563,0.9887459807073955,0.999844022143519,0.9987529997507149,0.9959043018953196,0.9886914378029079
+26,0.011765335466519483,0.058604288834275924,0.9962768657979354,0.9871382636655949,0.9998442385341351,0.9985795340320209,0.9962707650581987,0.9870550161812298
+27,0.011187340885168769,0.06167821835283298,0.9960512213008406,0.9871382636655949,0.9998823057803765,0.9983112939968224,0.9960458679319889,0.9870689655172413
+28,0.012367742936232329,0.06289260987779335,0.9957973712416088,0.9876741693461951,0.9998595163480708,0.9983233561825594,0.9957908415491963,0.9875876956287102
+29,0.010554071233931246,0.06150704976830069,0.9964460991707564,0.9876741693461951,0.9998957347274312,0.9983377159274844,0.9964416831403559,0.987601078167116
+30,0.011599305958529175,0.061056513950180774,0.9961922491115248,0.9876741693461951,0.9998390976658966,0.9983319720295143,0.9961865484025875,0.987601078167116
+31,0.010794812409661302,0.06143563166500288,0.9965871269814407,0.9876741693461951,0.9998747687042868,0.9983308232499204,0.996582596661677,0.987601078167116
+32,0.011800958064740092,0.05939523252835228,0.9962486602357985,0.9882100750267953,0.9998780941189012,0.9984008988051544,0.9962417700415384,0.9881465517241379
+33,0.010718018244260052,0.06093760498371155,0.9964178936086197,0.9876741693461951,0.9998868738499256,0.9983560964009884,0.9964143538778621,0.987601078167116
+34,0.010179058428227089,0.06254719717613753,0.9968691826028093,0.9876741693461951,0.9998930139336559,0.9983474805540336,0.9968643182010791,0.987601078167116
+35,0.010317418424680232,0.0629325036402684,0.9964460991707564,0.9871382636655949,0.9999166991594679,0.9983061244886495,0.9964416831403559,0.9870550161812298

training_metrics.csv

@@ -0,0 +1,36 @@
+epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1
+1,0.14566365511793838,0.08487027064205366,0.9427709144243245,0.9694533762057878,0.9851093454718796,0.994546168877493,0.9418307961354319,0.9689034369885434
+2,0.07756046253042306,0.07278770848006681,0.9740508828340949,0.977491961414791,0.9945806704518922,0.9966455635855249,0.9738085748448443,0.9775641025641025
+3,0.06155229155601513,0.06513803804419048,0.9785919783381283,0.9796355841371919,0.9965701100870393,0.9956719728796114,0.9784332111499446,0.9794372294372294
+4,0.05817294309795518,0.05396232030495187,0.9809048344333503,0.9828510182207931,0.9966998044392082,0.9976949737446423,0.980759939750476,0.9826839826839827
+5,0.05050496557747315,0.05946147611957654,0.9830484571557511,0.9828510182207931,0.9974789077145225,0.9974652178258427,0.9829392227552729,0.9827956989247312
+6,0.04554648937070848,0.07530270049713815,0.9841766796412252,0.9839228295819936,0.9980721664550192,0.997233164347855,0.9840828486310115,0.9836779107725789
+7,0.045750615091004296,0.07067370211100656,0.9848536131325097,0.9823151125401929,0.9979605836791916,0.9973560837644129,0.9847620669107006,0.9823245848955544
+8,0.041395390699091934,0.05189927040907731,0.9865741524228578,0.9839228295819936,0.9983463015433974,0.9980631576045188,0.9865102306863912,0.9838536060279871
+9,0.038652644766771084,0.06319661472004709,0.9872510859141422,0.9812433011789925,0.998584051186131,0.9982010111557986,0.9871918390478889,0.9809471965160588
+10,0.03613892484208536,0.059335076348957905,0.9882946917132058,0.9823151125401929,0.9987184138476197,0.9980861331963987,0.9882426268521404,0.9824374667376264
+11,0.03727046646559817,0.052120874501117936,0.9881254583403847,0.9844587352625938,0.9986679980162962,0.9983072732682435,0.98807264073434,0.984366576819407
+12,0.033966494260936154,0.04754341436831514,0.9886049528967112,0.9844587352625938,0.9988358900855194,0.9984238743970343,0.9885578339186587,0.9843497031840259
+13,0.031665333392081836,0.06848823487566982,0.9895639420093643,0.9817792068595927,0.9991228049491042,0.9981137039066548,0.9895255350469936,0.9817792068595927
+14,0.034051497119330254,0.07028256035694355,0.9884921306481638,0.9817792068595927,0.9989677658460311,0.9976398323241303,0.9884484711211778,0.9815016322089227
+15,0.03127622385937226,0.04463923775210641,0.9896485586957748,0.9855305466237942,0.9990817177808828,0.9985140535951632,0.9896089923270761,0.9854604200323102
+16,0.029261085257816156,0.06298187336737704,0.9902690810627855,0.9844587352625938,0.999124038057394,0.9980832612474138,0.9902379672335248,0.9843158464034614
+17,0.0289509687084124,0.07007141667642777,0.990184464376375,0.9844587352625938,0.9992388682835698,0.9978041078060722,0.9901561439239647,0.984400215169446
+18,0.029313751012666877,0.048654140910535,0.990381903311333,0.9871382636655949,0.9992080039808491,0.9981320843801588,0.9903473264075636,0.9870828848223897
+19,0.027540785092642957,0.05571309371680692,0.9907203700569752,0.9823151125401929,0.9993202518395994,0.9980108881329919,0.990686482661005,0.9822294022617124
+20,0.026049582501074288,0.051708202365893643,0.9913126868618491,0.9860664523043944,0.999335620346661,0.9976949737446423,0.9912861427035591,0.986021505376344
+21,0.024673762814315024,0.06106355182610907,0.991538331358944,0.9855305466237942,0.9994650585305271,0.9986467376382699,0.9915139171758316,0.9853975121687398
+22,0.016652436815372457,0.05243686088315927,0.9946409431939979,0.9876741693461951,0.9996888462052035,0.9988460508978287,0.9946318585070916,0.987601078167116
+23,0.015147703151754628,0.049660267170602485,0.9949512043775033,0.9882100750267953,0.9997919563337484,0.9988615594223478,0.994941073396829,0.9881465517241379
+24,0.013805163181409742,0.057504319061803665,0.9953178766852824,0.984994640943194,0.9998115826444022,0.9988247984753398,0.9953099395377748,0.9848648648648649
+25,0.012066704250700654,0.05450154134219099,0.9959101934901563,0.9887459807073955,0.999844022143519,0.9987529997507149,0.9959043018953196,0.9886914378029079
+26,0.011765335466519483,0.058604288834275924,0.9962768657979354,0.9871382636655949,0.9998442385341351,0.9985795340320209,0.9962707650581987,0.9870550161812298
+27,0.011187340885168769,0.06167821835283298,0.9960512213008406,0.9871382636655949,0.9998823057803765,0.9983112939968224,0.9960458679319889,0.9870689655172413
+28,0.012367742936232329,0.06289260987779335,0.9957973712416088,0.9876741693461951,0.9998595163480708,0.9983233561825594,0.9957908415491963,0.9875876956287102
+29,0.010554071233931246,0.06150704976830069,0.9964460991707564,0.9876741693461951,0.9998957347274312,0.9983377159274844,0.9964416831403559,0.987601078167116
+30,0.011599305958529175,0.061056513950180774,0.9961922491115248,0.9876741693461951,0.9998390976658966,0.9983319720295143,0.9961865484025875,0.987601078167116
+31,0.010794812409661302,0.06143563166500288,0.9965871269814407,0.9876741693461951,0.9998747687042868,0.9983308232499204,0.996582596661677,0.987601078167116
+32,0.011800958064740092,0.05939523252835228,0.9962486602357985,0.9882100750267953,0.9998780941189012,0.9984008988051544,0.9962417700415384,0.9881465517241379
+33,0.010718018244260052,0.06093760498371155,0.9964178936086197,0.9876741693461951,0.9998868738499256,0.9983560964009884,0.9964143538778621,0.987601078167116
+34,0.010179058428227089,0.06254719717613753,0.9968691826028093,0.9876741693461951,0.9998930139336559,0.9983474805540336,0.9968643182010791,0.987601078167116
+35,0.010317418424680232,0.0629325036402684,0.9964460991707564,0.9871382636655949,0.9999166991594679,0.9983061244886495,0.9964416831403559,0.9870550161812298