diff --git a/.gitattributes b/.gitattributes index a6344aac8c09253b3b630fb776ae94478aa0275b..d9e2184a393cd7a6a5b5ed1c6ac5331d9459c1bb 100644 --- a/.gitattributes +++ b/.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 diff --git a/README.md b/README.md new file mode 100644 index 0000000000000000000000000000000000000000..e8cf3ebc2d0c98fcc2eb6e179e20ef9eae4bf30d --- /dev/null +++ b/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/ diff --git a/config.json b/config.json new file mode 100644 index 0000000000000000000000000000000000000000..3335fe63317ccce99a14fb470118339b1c990f7a --- /dev/null +++ b/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" +} \ No newline at end of file diff --git a/confusion_matrices/PiT_Confusion_Matrix_a.png b/confusion_matrices/PiT_Confusion_Matrix_a.png new file mode 100644 index 0000000000000000000000000000000000000000..b357ab3ef1d80b365da69e87bcc30b22531a8377 Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_a.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_b.png b/confusion_matrices/PiT_Confusion_Matrix_b.png new file mode 100644 index 0000000000000000000000000000000000000000..8bf32f7b1d86d177c09370140dc57116af7faf4c Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_b.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_c.png b/confusion_matrices/PiT_Confusion_Matrix_c.png new file mode 100644 index 0000000000000000000000000000000000000000..d73b36a4f57ee7819aae9ae038edc7454f406985 Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_c.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_d.png b/confusion_matrices/PiT_Confusion_Matrix_d.png new file mode 100644 index 0000000000000000000000000000000000000000..0e2a13d27c976312038baa203cceb794645a294c Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_d.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_e.png b/confusion_matrices/PiT_Confusion_Matrix_e.png new file mode 100644 index 0000000000000000000000000000000000000000..9b344cfb5200a68e8d09125d054e09cb22440e89 Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_e.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_f.png b/confusion_matrices/PiT_Confusion_Matrix_f.png new file mode 100644 index 0000000000000000000000000000000000000000..cd0acbaf1711598d25e2f22e8b23303a5a66c42c Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_f.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_g.png b/confusion_matrices/PiT_Confusion_Matrix_g.png new file mode 100644 index 0000000000000000000000000000000000000000..df41e5b848ea28a93dcafa902108c7904b9c488e Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_g.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_h.png b/confusion_matrices/PiT_Confusion_Matrix_h.png new file mode 100644 index 0000000000000000000000000000000000000000..629df509d22b52efc9e07367b470ef4311e07979 Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_h.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_i.png b/confusion_matrices/PiT_Confusion_Matrix_i.png new file mode 100644 index 0000000000000000000000000000000000000000..33132dddfc78d1cf86fd6a833bd71bbf60c4fd91 Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_i.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_j.png b/confusion_matrices/PiT_Confusion_Matrix_j.png new file mode 100644 index 0000000000000000000000000000000000000000..f6e20d00c0bfa1f7f74c60a0b0d68fa089c23f06 Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_j.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_k.png b/confusion_matrices/PiT_Confusion_Matrix_k.png new file mode 100644 index 0000000000000000000000000000000000000000..9e6e98680db48faecc5b2b64ac803d4ab020d11b Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_k.png differ diff --git a/confusion_matrices/PiT_Confusion_Matrix_l.png b/confusion_matrices/PiT_Confusion_Matrix_l.png new file mode 100644 index 0000000000000000000000000000000000000000..b6c819929c055318ec172d812e45fe5c2ead23dd Binary files /dev/null and b/confusion_matrices/PiT_Confusion_Matrix_l.png differ diff --git a/evaluation_results.csv b/evaluation_results.csv new file mode 100644 index 0000000000000000000000000000000000000000..5ec7ba637bb51c02b5fcd121d5b9e2fd21ac8e7b --- /dev/null +++ b/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 diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..ae1b03cbdfa4516893166eac05a7708a821c25b9 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b21dce370730d57725c0300c4ef486825afacdbbf82ccc501d031fd1139d03f9 +size 290985688 diff --git a/pit-gravit-b3.pth b/pit-gravit-b3.pth new file mode 100644 index 0000000000000000000000000000000000000000..6f31b8e356e09b4864a022b2e5b72154edc6727b --- /dev/null +++ b/pit-gravit-b3.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01e02d406a585b90d523b4f5804cb44e7f626aedea9d00507365e2e56c3d1129 +size 291039882 diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6f31b8e356e09b4864a022b2e5b72154edc6727b --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01e02d406a585b90d523b4f5804cb44e7f626aedea9d00507365e2e56c3d1129 +size 291039882 diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_a.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_a.png new file mode 100644 index 0000000000000000000000000000000000000000..a6b1f2a7647c67600023f624634cadf65e132477 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_a.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_b.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_b.png new file mode 100644 index 0000000000000000000000000000000000000000..6e7798b8050b078535f058fa66fe5edeeb97f91b Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_b.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_c.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_c.png new file mode 100644 index 0000000000000000000000000000000000000000..c1b32442437b6ee74625b809ca6c63d383cac3ca Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_c.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_d.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_d.png new file mode 100644 index 0000000000000000000000000000000000000000..d74e42faa5ea2ac525cb66ab68e510df7700286f Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_d.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_e.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_e.png new file mode 100644 index 0000000000000000000000000000000000000000..870f4a2f4f4d2cdf4e0295341b960dd9a8b27510 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_e.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_f.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_f.png new file mode 100644 index 0000000000000000000000000000000000000000..670d75b5d0d46c7553aa03ee4ffd275e34b8d303 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_f.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_g.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_g.png new file mode 100644 index 0000000000000000000000000000000000000000..f2efb2064a06be38367f0284ad68641094f707c8 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_g.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_h.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_h.png new file mode 100644 index 0000000000000000000000000000000000000000..111aa8dd0a7e716d3d618b21749ec30071689b26 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_h.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_i.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_i.png new file mode 100644 index 0000000000000000000000000000000000000000..9aebb5f4afa0b4a263ed03269009d2df28228682 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_i.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_j.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_j.png new file mode 100644 index 0000000000000000000000000000000000000000..79378b02db1da817fad0cd0145c92eb933e7e643 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_j.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_k.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_k.png new file mode 100644 index 0000000000000000000000000000000000000000..a4fa7f6138bd566af9360edbd6d6bb5261d22891 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_k.png differ diff --git a/roc_confusion_matrix/PiT_roc_confusion_matrix_l.png b/roc_confusion_matrix/PiT_roc_confusion_matrix_l.png new file mode 100644 index 0000000000000000000000000000000000000000..03c4177eae6c59f67638bcff532dadc496975540 Binary files /dev/null and b/roc_confusion_matrix/PiT_roc_confusion_matrix_l.png differ diff --git a/roc_curves/PiT_ROC_a.png b/roc_curves/PiT_ROC_a.png new file mode 100644 index 0000000000000000000000000000000000000000..c852b12ac44d519474bdbfe980a1f4272967b36e Binary files /dev/null and b/roc_curves/PiT_ROC_a.png differ diff --git a/roc_curves/PiT_ROC_b.png b/roc_curves/PiT_ROC_b.png new file mode 100644 index 0000000000000000000000000000000000000000..d88313a05eaaffbdfa1c4335e3aa8c0a6e3ec364 Binary files /dev/null and b/roc_curves/PiT_ROC_b.png differ diff --git a/roc_curves/PiT_ROC_c.png b/roc_curves/PiT_ROC_c.png new file mode 100644 index 0000000000000000000000000000000000000000..b8332fc2b71bb4019a8d882a370d76769885f3e5 Binary files /dev/null and b/roc_curves/PiT_ROC_c.png differ diff --git a/roc_curves/PiT_ROC_d.png b/roc_curves/PiT_ROC_d.png new file mode 100644 index 0000000000000000000000000000000000000000..649ea1b8e1c2b147316c358b7c1b7b46cc113cd0 Binary files /dev/null and b/roc_curves/PiT_ROC_d.png differ diff --git a/roc_curves/PiT_ROC_e.png b/roc_curves/PiT_ROC_e.png new file mode 100644 index 0000000000000000000000000000000000000000..de7403f12ed4936e9f859686a1a72ffc1f6f2f6b Binary files /dev/null and b/roc_curves/PiT_ROC_e.png differ diff --git a/roc_curves/PiT_ROC_f.png b/roc_curves/PiT_ROC_f.png new file mode 100644 index 0000000000000000000000000000000000000000..ed3fc80f4adbed86c8129784da043a3a61bc602e Binary files /dev/null and b/roc_curves/PiT_ROC_f.png differ diff --git a/roc_curves/PiT_ROC_g.png b/roc_curves/PiT_ROC_g.png new file mode 100644 index 0000000000000000000000000000000000000000..32df596b285b29d81de45d1d1115b9e364a6a127 Binary files /dev/null and b/roc_curves/PiT_ROC_g.png differ diff --git a/roc_curves/PiT_ROC_h.png b/roc_curves/PiT_ROC_h.png new file mode 100644 index 0000000000000000000000000000000000000000..5f6bedebd9aa2cb592a3c6db98a233954086480b Binary files /dev/null and b/roc_curves/PiT_ROC_h.png differ diff --git a/roc_curves/PiT_ROC_i.png b/roc_curves/PiT_ROC_i.png new file mode 100644 index 0000000000000000000000000000000000000000..e77c495e7f68423d8a07ee23959403c7902034cd Binary files /dev/null and b/roc_curves/PiT_ROC_i.png differ diff --git a/roc_curves/PiT_ROC_j.png b/roc_curves/PiT_ROC_j.png new file mode 100644 index 0000000000000000000000000000000000000000..638cd52874a551ba08c035cb9342f9b13ba52d56 Binary files /dev/null and b/roc_curves/PiT_ROC_j.png differ diff --git a/roc_curves/PiT_ROC_k.png b/roc_curves/PiT_ROC_k.png new file mode 100644 index 0000000000000000000000000000000000000000..4201c3a55bc8ffb87029ff92e111f7df1d1b395a Binary files /dev/null and b/roc_curves/PiT_ROC_k.png differ diff --git a/roc_curves/PiT_ROC_l.png b/roc_curves/PiT_ROC_l.png new file mode 100644 index 0000000000000000000000000000000000000000..7c141bbdf71393ee3507581d320e729cf63dcc4e Binary files /dev/null and b/roc_curves/PiT_ROC_l.png differ diff --git a/training_curves/PiT_accuracy.png b/training_curves/PiT_accuracy.png new file mode 100644 index 0000000000000000000000000000000000000000..8572c9095f961c1bae9d57b9d9c2ea009b517b36 Binary files /dev/null and b/training_curves/PiT_accuracy.png differ diff --git a/training_curves/PiT_auc.png b/training_curves/PiT_auc.png new file mode 100644 index 0000000000000000000000000000000000000000..ed1c4cec0eb1048d296334af1ddd8763d78b8708 Binary files /dev/null and b/training_curves/PiT_auc.png differ diff --git a/training_curves/PiT_combined_metrics.png b/training_curves/PiT_combined_metrics.png new file mode 100644 index 0000000000000000000000000000000000000000..2983c6da8010b623148a8738f065864dd2795b2a --- /dev/null +++ b/training_curves/PiT_combined_metrics.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6167b8de237e51c3a23135c6703a089f04e96bd2d58b22d881859fa41d426660 +size 147778 diff --git a/training_curves/PiT_f1.png b/training_curves/PiT_f1.png new file mode 100644 index 0000000000000000000000000000000000000000..ef070255b1dcf7cffd1ee64dfe44eb8da5ec7075 Binary files /dev/null and b/training_curves/PiT_f1.png differ diff --git a/training_curves/PiT_loss.png b/training_curves/PiT_loss.png new file mode 100644 index 0000000000000000000000000000000000000000..8d8fa5a30fd663fb766885f4e3642755540f9104 Binary files /dev/null and b/training_curves/PiT_loss.png differ diff --git a/training_curves/PiT_metrics.csv b/training_curves/PiT_metrics.csv new file mode 100644 index 0000000000000000000000000000000000000000..0bfa3a0f7ac5b7802ee3c9a053fce1b69878b053 --- /dev/null +++ b/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 diff --git a/training_metrics.csv b/training_metrics.csv new file mode 100644 index 0000000000000000000000000000000000000000..0bfa3a0f7ac5b7802ee3c9a053fce1b69878b053 --- /dev/null +++ b/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 diff --git a/training_notebook_b3.ipynb b/training_notebook_b3.ipynb new file mode 100644 index 0000000000000000000000000000000000000000..701938a378aeab8da8272176e7836018be7f9ab9 --- /dev/null +++ b/training_notebook_b3.ipynb @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3fef560bd8b2a519a8ce42ba3abdb2129902597df1f704f87faaad9d586495a8 +size 21312302