diff --git a/.gitattributes b/.gitattributes index a6344aac8c09253b3b630fb776ae94478aa0275b..24b5b60305996d06754f4374cef7a7cb5a167278 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/CvT_combined_metrics.png filter=lfs diff=lfs merge=lfs -text +training_notebook_c2.ipynb filter=lfs diff=lfs merge=lfs -text diff --git a/README.md b/README.md new file mode 100644 index 0000000000000000000000000000000000000000..d440fdf513c8491a3f70d0ce64b86ed6dc86c796 --- /dev/null +++ b/README.md @@ -0,0 +1,166 @@ +--- +license: apache-2.0 +tags: +- image-classification +- pytorch +- timm +- cvt +- vision-transformer +- transformer +- gravitational-lensing +- strong-lensing +- astronomy +- astrophysics +datasets: +- parlange/gravit-c21-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: CvT-c2 + 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.6983 + name: Average Accuracy + - type: auc + value: 0.7470 + name: Average AUC-ROC + - type: f1 + value: 0.4396 + name: Average F1-Score +--- + +# 🌌 cvt-gravit-c2 + +πŸ”­ 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**: CvT +- **πŸ§ͺ Experiment**: C2 - C21+J24-half +- **🌌 Dataset**: C21+J24 +- **πŸͺ Fine-tuning Strategy**: half + + + +## πŸ’» Quick Start + +```python +import torch +import timm + +# Load the model directly from the Hub +model = timm.create_model( + 'hf-hub:parlange/cvt-gravit-c2', + 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:** C21+J24 (CaΓ±ameras et al. 2021 + Jaelani et al. 2024) +**Fine-tuning Strategy:** half + + +| πŸ”§ Parameter | πŸ“ Value | +|--------------|----------| +| Batch Size | 192 | +| Learning Rate | AdamW with ReduceLROnPlateau | +| Epochs | 100 | +| Patience | 10 | +| Optimizer | AdamW | +| Scheduler | ReduceLROnPlateau | +| Image Size | 224x224 | +| Fine Tune Mode | half | +| Stochastic Depth Probability | 0.1 | + + +## πŸ“ˆ Training Curves + +![Combined Training Metrics](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/training_curves/CvT_combined_metrics.png) + + +## 🏁 Final Epoch Training Metrics + +| Metric | Training | Validation | +|:---------:|:-----------:|:-------------:| +| πŸ“‰ Loss | 0.3623 | 0.3242 | +| 🎯 Accuracy | 0.8036 | 0.8513 | +| πŸ“Š AUC-ROC | 0.9105 | 0.9527 | +| βš–οΈ F1 Score | 0.8057 | 0.8623 | + + +## β˜‘οΈ 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/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_a.png) +![ROC + Confusion Matrix - Dataset B](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_b.png) +![ROC + Confusion Matrix - Dataset C](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_c.png) +![ROC + Confusion Matrix - Dataset D](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_d.png) +![ROC + Confusion Matrix - Dataset E](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_e.png) +![ROC + Confusion Matrix - Dataset F](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_f.png) +![ROC + Confusion Matrix - Dataset G](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_g.png) +![ROC + Confusion Matrix - Dataset H](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_h.png) +![ROC + Confusion Matrix - Dataset I](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_i.png) +![ROC + Confusion Matrix - Dataset J](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_j.png) +![ROC + Confusion Matrix - Dataset K](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_roc_confusion_matrix_k.png) +![ROC + Confusion Matrix - Dataset L](https://huggingface.co/parlange/cvt-gravit-c2/resolve/main/roc_confusion_matrix/CvT_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.6983 | +| πŸ“ˆ Average AUC-ROC | 0.7470 | +| βš–οΈ Average F1-Score | 0.4396 | + + +## πŸ“˜ 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..f19afc67db088a1cc2047462b1d09315cbb8bd49 --- /dev/null +++ b/config.json @@ -0,0 +1,76 @@ +{ + "architecture": "cvt_13_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_c2", + "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": "cvt_gravit_c2", + "experiment": "c2", + "training_strategy": "half", + "dataset": "C21+J24", + "hyperparameters": { + "batch_size": "192", + "learning_rate": "AdamW with ReduceLROnPlateau", + "epochs": "100", + "patience": "10", + "optimizer": "AdamW", + "scheduler": "ReduceLROnPlateau", + "image_size": "224x224", + "fine_tune_mode": "half", + "stochastic_depth_probability": "0.1" + }, + "hf_hub_id": "parlange/cvt-gravit-c2", + "license": "apache-2.0" +} \ No newline at end of file diff --git a/confusion_matrices/CvT_Confusion_Matrix_a.png b/confusion_matrices/CvT_Confusion_Matrix_a.png new file mode 100644 index 0000000000000000000000000000000000000000..32b3087b6d59f53a96589699367b6b0d35255d3b Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_a.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_b.png b/confusion_matrices/CvT_Confusion_Matrix_b.png new file mode 100644 index 0000000000000000000000000000000000000000..fe98490b0a3cfc95f186dfd70e353ae74ea1ad88 Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_b.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_c.png b/confusion_matrices/CvT_Confusion_Matrix_c.png new file mode 100644 index 0000000000000000000000000000000000000000..40fb248e1ea965e575851d9fb7c342fafada684b Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_c.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_d.png b/confusion_matrices/CvT_Confusion_Matrix_d.png new file mode 100644 index 0000000000000000000000000000000000000000..8b874031a091a488948f75b0bbf5988caf75848d Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_d.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_e.png b/confusion_matrices/CvT_Confusion_Matrix_e.png new file mode 100644 index 0000000000000000000000000000000000000000..b0321dfb143037de28fcb39750189565de347ab6 Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_e.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_f.png b/confusion_matrices/CvT_Confusion_Matrix_f.png new file mode 100644 index 0000000000000000000000000000000000000000..f04cb1f3301d0a984f64736cb246dda0b6b4c539 Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_f.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_g.png b/confusion_matrices/CvT_Confusion_Matrix_g.png new file mode 100644 index 0000000000000000000000000000000000000000..3e282d6615387375cd5778e17e82025c7b47fcc6 Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_g.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_h.png b/confusion_matrices/CvT_Confusion_Matrix_h.png new file mode 100644 index 0000000000000000000000000000000000000000..e8c0b8ccb3c0adff954182945976a9a7ba37d399 Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_h.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_i.png b/confusion_matrices/CvT_Confusion_Matrix_i.png new file mode 100644 index 0000000000000000000000000000000000000000..3bd9e9355b4cd0f277fa83a025ce6ccc9b896b0b Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_i.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_j.png b/confusion_matrices/CvT_Confusion_Matrix_j.png new file mode 100644 index 0000000000000000000000000000000000000000..9873f3c797caa89a8c56adbe4815b3126ed20bfe Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_j.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_k.png b/confusion_matrices/CvT_Confusion_Matrix_k.png new file mode 100644 index 0000000000000000000000000000000000000000..a04a63327a428ea9e8dbcbd45b6883f01ca4b0fc Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_k.png differ diff --git a/confusion_matrices/CvT_Confusion_Matrix_l.png b/confusion_matrices/CvT_Confusion_Matrix_l.png new file mode 100644 index 0000000000000000000000000000000000000000..82447a0b2fbc6439ccf5d8756a9b6c84a7fb8380 Binary files /dev/null and b/confusion_matrices/CvT_Confusion_Matrix_l.png differ diff --git a/cvt-gravit-c2.pth b/cvt-gravit-c2.pth new file mode 100644 index 0000000000000000000000000000000000000000..363636a9336523e33de6c8bfa120c98dfebf1c4e --- /dev/null +++ b/cvt-gravit-c2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e003815d79d6f2a95583e2f46e27a357d42c225b8e9b6448e6464767bca64c9d +size 125471131 diff --git a/evaluation_results.csv b/evaluation_results.csv new file mode 100644 index 0000000000000000000000000000000000000000..51ad31002d20e48ac3ebf8b61b5b308068673ca5 --- /dev/null +++ b/evaluation_results.csv @@ -0,0 +1,133 @@ +Model,Dataset,Loss,Accuracy,AUCROC,F1 +ViT,a,0.35447569389258865,0.8949115044247787,0.9020846228498507,0.7480106100795756 +ViT,b,0.2228443425890036,0.9264382269726501,0.9263609576427256,0.5465116279069767 +ViT,c,0.46229862214933587,0.8349575605155611,0.8684438305709024,0.34944237918215615 +ViT,d,0.11673173789463115,0.9537881169443572,0.9704917127071824,0.6573426573426573 +ViT,e,0.3652098159562351,0.8825466520307355,0.920767426019829,0.7249357326478149 +ViT,f,0.24608832064126923,0.9108519842016175,0.9203345361214339,0.22926829268292684 +ViT,g,0.10483654439449311,0.9635,0.997473,0.9644999189495866 +ViT,h,0.23178722894191742,0.915,0.9939590555555555,0.9210526315789473 +ViT,i,0.04857917896906535,0.978,0.9990572222222223,0.9782966129562644 +ViT,j,2.494326035181681,0.6106666666666667,0.5831323333333334,0.42349457058242845 +ViT,k,2.4380686638752618,0.6251666666666666,0.7805802777777779,0.43278688524590164 +ViT,l,1.0272723743838732,0.8127329565949261,0.7993805230717175,0.7184308053873272 +MLP-Mixer,a,1.230455079964832,0.6227876106194691,0.8958911227772556,0.49028400597907323 +MLP-Mixer,b,1.0728926989350893,0.7004086765168186,0.9182900552486188,0.25604996096799376 +MLP-Mixer,c,1.374837134027586,0.5576862621817039,0.8979152854511969,0.18904899135446687 +MLP-Mixer,d,0.09552026474693218,0.9603898145237346,0.9868913443830571,0.7224669603524229 +MLP-Mixer,e,0.9593323631422711,0.7069154774972558,0.9188677817301143,0.5512605042016807 +MLP-Mixer,f,0.9257462782946794,0.7154410381794245,0.9306221006103087,0.09779367918902802 +MLP-Mixer,g,0.5643243643840155,0.8425,0.991425611111111,0.8635773061931572 +MLP-Mixer,h,0.7244052359660467,0.7668333333333334,0.9891666111111111,0.8104592873594364 +MLP-Mixer,i,0.04615406060218811,0.9803333333333333,0.9994367777777778,0.980655737704918 +MLP-Mixer,j,3.0292422666549683,0.45216666666666666,0.392282,0.28309705561613957 +MLP-Mixer,k,2.5110719747940697,0.59,0.7661271111111111,0.3453964874933475 +MLP-Mixer,l,1.4846716919555334,0.6762053625105207,0.7295511702036557,0.5855010004617516 +CvT,a,0.7465745627352621,0.6493362831858407,0.7317079694031161,0.4389380530973451 +CvT,b,0.7336456650122649,0.6765168186104998,0.7552670349907918,0.1942051683633516 +CvT,c,0.8642418710588097,0.5919522162841874,0.6964806629834255,0.16041397153945666 +CvT,d,0.06205783033066015,0.9761081420936812,0.9876427255985267,0.7654320987654321 +CvT,e,0.6019917449757506,0.7178924259055982,0.7936123514720351,0.4910891089108911 +CvT,f,0.5685286294680824,0.7414895617829603,0.8061353821076506,0.08274941608274941 +CvT,g,0.4509977758725484,0.8055,0.9201512777777776,0.8277999114652501 +CvT,h,0.5202355206807454,0.7606666666666667,0.9072719444444444,0.7961964235026966 +CvT,i,0.09494428576032321,0.9643333333333334,0.9977035555555557,0.9632554945054945 +CvT,j,2.988422914981842,0.3456666666666667,0.14668444444444442,0.022896963663514187 +CvT,k,2.6323694267769655,0.5045,0.6181494444444444,0.0300163132137031 +CvT,l,1.337245315202257,0.645425033064807,0.6032419706344807,0.5021944632005402 +Swin,a,0.47572549887463056,0.8407079646017699,0.905882487792577,0.6742081447963801 +Swin,b,0.24361524523634911,0.9163784973278843,0.9362615101289135,0.5283687943262412 +Swin,c,0.4370936370240709,0.8535051870480981,0.9087605893186003,0.3900523560209424 +Swin,d,0.038348094671021904,0.9880540710468406,0.9911620626151013,0.8869047619047619 +Swin,e,0.3579506372581067,0.8781558726673985,0.9260273972602739,0.7286063569682152 +Swin,f,0.24650774364781286,0.9156479217603912,0.9413092437445593,0.24937238493723848 +Swin,g,0.11494702147444089,0.9593333333333334,0.9989898888888888,0.9607969151670951 +Swin,h,0.2175228010714054,0.926,0.9979807777777777,0.9308841843088418 +Swin,i,0.006121216081082821,0.9973333333333333,0.9999798888888889,0.9973315543695798 +Swin,j,2.5422419211069744,0.5825,0.4893003333333333,0.3679031037093111 +Swin,k,2.433416116627554,0.6205,0.7913794999999999,0.39036144578313253 +Swin,l,1.035569912688268,0.8089455332451605,0.7797953948083542,0.7088143668682426 +CaiT,a,0.3509529214517205,0.9081858407079646,0.8966973093999068,0.7726027397260274 +CaiT,b,0.1907231829655279,0.9380697893744105,0.9234548802946593,0.5887265135699373 +CaiT,c,0.3048490960337163,0.90883370009431,0.8791160220994475,0.493006993006993 +CaiT,d,0.06549901952829443,0.9849104055328513,0.969243093922652,0.8545454545454545 +CaiT,e,0.31167979835318943,0.9187705817782656,0.9264058124574283,0.7921348314606742 +CaiT,f,0.1541684599891403,0.9499717886025955,0.9222261921687871,0.3464373464373464 +CaiT,g,0.07805611325552066,0.9708333333333333,0.9986172777777778,0.9714937286202965 +CaiT,h,0.13856186520308256,0.9553333333333334,0.997130611111111,0.9569961489088575 +CaiT,i,0.011666435472667217,0.9956666666666667,0.9999013333333333,0.9956594323873121 +CaiT,j,1.8389671653707822,0.6116666666666667,0.7423962222222222,0.4151606425702811 +CaiT,k,1.7725774958133698,0.6365,0.8888650555555555,0.4312907431551499 +CaiT,l,0.7395369254032035,0.8362991463268006,0.8693810723675515,0.7436693965922997 +DeiT,a,0.48058320357736234,0.8263274336283186,0.8941450218931248,0.6594360086767896 +DeiT,b,0.23002449519573911,0.9251807607670544,0.9313581952117864,0.5608856088560885 +DeiT,c,0.49494195908204974,0.8154668343288274,0.8907605893186004,0.34118967452300786 +DeiT,d,0.05036040664735698,0.9849104055328513,0.9769023941068141,0.8636363636363636 +DeiT,e,0.338863200106291,0.8792535675082327,0.9161961704382048,0.7342995169082126 +DeiT,f,0.26403015722496653,0.9037050968591311,0.9291450866890099,0.2289156626506024 +DeiT,g,0.10851164469867945,0.9641666666666666,0.9990410000000001,0.9653393519264872 +DeiT,h,0.2489620513096452,0.906,0.9981344444444444,0.9139194139194139 +DeiT,i,0.013259729760388533,0.9958333333333333,0.9998315555555556,0.9958423415932147 +DeiT,j,1.2026229511300723,0.7143333333333334,0.7246498888888889,0.6356292517006803 +DeiT,k,1.1073710439900557,0.746,0.8698901111111111,0.6623836951705804 +DeiT,l,0.5658274294531473,0.8476012985451485,0.867833726587774,0.7854785478547854 +DeiT3,a,0.39277621998196155,0.8661504424778761,0.9195532732705195,0.7125890736342043 +DeiT3,b,0.338128161960636,0.8824269097767997,0.9331012891344382,0.44510385756676557 +DeiT3,c,0.323060417608134,0.8883998742533794,0.922292817679558,0.4580152671755725 +DeiT3,d,0.12409640010358478,0.9553599497013517,0.9608121546961326,0.6787330316742082 +DeiT3,e,0.24973662732461413,0.9209659714599341,0.9483084840687203,0.8064516129032258 +DeiT3,f,0.2540075041596123,0.9116042881324055,0.9380772021883802,0.24193548387096775 +DeiT3,g,0.1656125110021482,0.9416666666666667,0.9990236666666666,0.944760101010101 +DeiT3,h,0.15762409150910875,0.9448333333333333,0.9990646111111111,0.9476017096723128 +DeiT3,i,0.05214000094247361,0.9803333333333333,0.9997376666666667,0.9806684141546527 +DeiT3,j,1.1591287109454473,0.696,0.7744774999999999,0.6248457424928013 +DeiT3,k,1.0456561943689981,0.7346666666666667,0.845634,0.6561555075593952 +DeiT3,l,0.5223108836063022,0.854033906456655,0.8898184372191467,0.7933968686181075 +Twins_SVT,a,0.4211153812640536,0.8307522123893806,0.8825833123189902,0.6433566433566433 +Twins_SVT,b,0.3625493723054758,0.8550770198050928,0.8962191528545118,0.37449118046132973 +Twins_SVT,c,0.47319920195681764,0.7868594781515247,0.8548139963167587,0.2893081761006289 +Twins_SVT,d,0.1203458983801289,0.9783087079534738,0.9818324125230202,0.8 +Twins_SVT,e,0.5213294555274637,0.7486278814489572,0.8316203738742148,0.5465346534653466 +Twins_SVT,f,0.3335461875583885,0.8666541282678202,0.9034523383543173,0.16292798110979928 +Twins_SVT,g,0.2639119902451833,0.9085,0.9744078888888889,0.912676952441546 +Twins_SVT,h,0.32257486327489215,0.8723333333333333,0.9662636666666669,0.8822263222632226 +Twins_SVT,i,0.13550377811988196,0.9738333333333333,0.9972788888888889,0.9733672603901612 +Twins_SVT,j,1.2430085968176523,0.49,0.43771377777777776,0.1896186440677966 +Twins_SVT,k,1.1146003757913907,0.5553333333333333,0.7234002222222222,0.2115839243498818 +Twins_SVT,l,0.6286477774643219,0.7480461704941685,0.7275090480198628,0.6162439337057046 +Twins_PCPVT,a,0.45601994748664115,0.7699115044247787,0.8394007473464615,0.5458515283842795 +Twins_PCPVT,b,0.3125818614145001,0.8773970449544168,0.9010699815837937,0.390625 +Twins_PCPVT,c,0.5049686531944119,0.7500785916378497,0.8135911602209945,0.23923444976076555 +Twins_PCPVT,d,0.3149096430453517,0.8918579063187677,0.9015690607734806,0.4208754208754209 +Twins_PCPVT,e,0.42039827045572575,0.8079034028540066,0.8655339438431847,0.5882352941176471 +Twins_PCPVT,f,0.3770137148085496,0.8412638706037239,0.8693597175042401,0.12899896800825594 +Twins_PCPVT,g,0.2785677030881246,0.9015,0.9626754444444443,0.9027480664801711 +Twins_PCPVT,h,0.3805647597312927,0.834,0.928301,0.8463437210737427 +Twins_PCPVT,i,0.2798018006483714,0.9091666666666667,0.9656723333333334,0.9096335599403084 +Twins_PCPVT,j,0.614702238559723,0.6835,0.7995154444444446,0.6018033130635353 +Twins_PCPVT,k,0.6159363424777985,0.6911666666666667,0.7903985,0.6076646199449502 +Twins_PCPVT,l,0.45535326129802217,0.7889864133702056,0.8498913163479216,0.7103004291845494 +PiT,a,0.3937257931823224,0.8296460176991151,0.8874127904755356,0.641860465116279 +PiT,b,0.2796248870145521,0.8777114115058158,0.91848802946593,0.4150375939849624 +PiT,c,0.5313189482209218,0.7613957874882112,0.8498581952117863,0.26666666666666666 +PiT,d,0.049343678185640734,0.9798805407104684,0.9911620626151012,0.8117647058823529 +PiT,e,0.3259278782832505,0.8518111964873765,0.9145841216983274,0.6715328467153284 +PiT,f,0.2841162405192056,0.8750235094978371,0.9172267022129574,0.17196261682242991 +PiT,g,0.1590204114516576,0.9338333333333333,0.9916004444444445,0.9369340746624305 +PiT,h,0.2924602138201396,0.8721666666666666,0.981646111111111,0.8849212303075769 +PiT,i,0.03693298858900865,0.988,0.999485,0.9879396984924623 +PiT,j,2.9977854507366817,0.461,0.277717,0.06477732793522267 +PiT,k,2.8756980224698783,0.5151666666666667,0.7229978888888889,0.07149696776252792 +PiT,l,1.2244331041709067,0.7434170975111218,0.6790239785353327,0.599849990624414 +Ensemble,a,,0.9070796460176991,0.941851401847734,0.79 +Ensemble,b,,0.9374410562716127,0.9600349907918969,0.6135922330097088 +Ensemble,c,,0.895001571832757,0.9307624309392265,0.48615384615384616 +Ensemble,d,,0.9911977365608299,0.9944677716390424,0.9186046511627907 +Ensemble,e,,0.9264544456641054,0.955384848255506,0.825065274151436 +Ensemble,f,,0.941696445363927,0.9599335198386041,0.33760683760683763 +Ensemble,g,,0.9701666666666666,0.9990522222222222,0.9710027539283979 +Ensemble,h,,0.9476666666666667,0.9979163333333333,0.9502219403931516 +Ensemble,i,,0.9986666666666667,0.9999886666666667,0.9986671109630123 +Ensemble,j,,0.5698333333333333,0.6426453333333333,0.31556616282153277 +Ensemble,k,,0.5983333333333334,0.8897323333333333,0.33055555555555555 +Ensemble,l,,0.8179632078874595,0.832089495815299,0.712386018237082 diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..df383a5e7147dbffef4d36de4bdae1d65a56f906 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:be02cc6e045512c352a07c5c5d91d58e636462c47757069251e2db386985c120 +size 125239576 diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..363636a9336523e33de6c8bfa120c98dfebf1c4e --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e003815d79d6f2a95583e2f46e27a357d42c225b8e9b6448e6464767bca64c9d +size 125471131 diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_a.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_a.png new file mode 100644 index 0000000000000000000000000000000000000000..b9ef22b293f6c209c42a35c4356621cbea459aec Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_a.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_b.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_b.png new file mode 100644 index 0000000000000000000000000000000000000000..2275c9c21fba6da0bac15f3a57472bd18ebc45d4 Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_b.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_c.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_c.png new file mode 100644 index 0000000000000000000000000000000000000000..f0a106e5c0adf1bfe90b006d77afee7295df90a1 Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_c.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_d.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_d.png new file mode 100644 index 0000000000000000000000000000000000000000..0adb3d418b92a975fd4abdee1d5998861366fdb2 Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_d.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_e.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_e.png new file mode 100644 index 0000000000000000000000000000000000000000..747f7d1ef0984abd8f399eeed097fa6c7bb5eb89 Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_e.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_f.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_f.png new file mode 100644 index 0000000000000000000000000000000000000000..b323025e32bc6f19138ca467d0c534c86d55a662 Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_f.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_g.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_g.png new file mode 100644 index 0000000000000000000000000000000000000000..14afdd03aa7bea7bea0eb5192269c4f5fb72a61b Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_g.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_h.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_h.png new file mode 100644 index 0000000000000000000000000000000000000000..cc05b3919b5d00b91c8d39cc305db62a16fe4c9d Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_h.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_i.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_i.png new file mode 100644 index 0000000000000000000000000000000000000000..3d73a4f9b9891f2e3efbf6010b8ca7b997ce9df5 Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_i.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_j.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_j.png new file mode 100644 index 0000000000000000000000000000000000000000..4200b1ddbcbf371c89d9abc661b0a863119e265e Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_j.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_k.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_k.png new file mode 100644 index 0000000000000000000000000000000000000000..139e04fc326edae122a70d643a88f31fde2da548 Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_k.png differ diff --git a/roc_confusion_matrix/CvT_roc_confusion_matrix_l.png b/roc_confusion_matrix/CvT_roc_confusion_matrix_l.png new file mode 100644 index 0000000000000000000000000000000000000000..47c36a9d571e2b1a8f8a86fabce44d9ac6efb894 Binary files /dev/null and b/roc_confusion_matrix/CvT_roc_confusion_matrix_l.png differ diff --git a/roc_curves/CvT_ROC_a.png b/roc_curves/CvT_ROC_a.png new file mode 100644 index 0000000000000000000000000000000000000000..b224700a94bed1a778d63451c0dedb28647d9eee Binary files /dev/null and b/roc_curves/CvT_ROC_a.png differ diff --git a/roc_curves/CvT_ROC_b.png b/roc_curves/CvT_ROC_b.png new file mode 100644 index 0000000000000000000000000000000000000000..49f72ffdb986822e6a56d35b3c560563249f4343 Binary files /dev/null and b/roc_curves/CvT_ROC_b.png differ diff --git a/roc_curves/CvT_ROC_c.png b/roc_curves/CvT_ROC_c.png new file mode 100644 index 0000000000000000000000000000000000000000..be3feed01faa7c19f5c0d33aeb75d20d9743f186 Binary files /dev/null and b/roc_curves/CvT_ROC_c.png differ diff --git a/roc_curves/CvT_ROC_d.png b/roc_curves/CvT_ROC_d.png new file mode 100644 index 0000000000000000000000000000000000000000..6cf0347c63c952f8153dddae3d69be9f05facb18 Binary files /dev/null and b/roc_curves/CvT_ROC_d.png differ diff --git a/roc_curves/CvT_ROC_e.png b/roc_curves/CvT_ROC_e.png new file mode 100644 index 0000000000000000000000000000000000000000..14c953ad33c6bf2861cbbe2f060450463c3e4e68 Binary files /dev/null and b/roc_curves/CvT_ROC_e.png differ diff --git a/roc_curves/CvT_ROC_f.png b/roc_curves/CvT_ROC_f.png new file mode 100644 index 0000000000000000000000000000000000000000..ae33487d6d32b39c67c5c4d72c435df7e6f8aa71 Binary files /dev/null and b/roc_curves/CvT_ROC_f.png differ diff --git a/roc_curves/CvT_ROC_g.png b/roc_curves/CvT_ROC_g.png new file mode 100644 index 0000000000000000000000000000000000000000..55d347b57e6cdf1b871034d82c570ff1d3911475 Binary files /dev/null and b/roc_curves/CvT_ROC_g.png differ diff --git a/roc_curves/CvT_ROC_h.png b/roc_curves/CvT_ROC_h.png new file mode 100644 index 0000000000000000000000000000000000000000..ae18a5072596d8cda01b4d3240c487c957bd4184 Binary files /dev/null and b/roc_curves/CvT_ROC_h.png differ diff --git a/roc_curves/CvT_ROC_i.png b/roc_curves/CvT_ROC_i.png new file mode 100644 index 0000000000000000000000000000000000000000..039a236f2da363c3f46c02108c8f4d33739d9014 Binary files /dev/null and b/roc_curves/CvT_ROC_i.png differ diff --git a/roc_curves/CvT_ROC_j.png b/roc_curves/CvT_ROC_j.png new file mode 100644 index 0000000000000000000000000000000000000000..e89dcab99e88c5863acabe647155368f3295cce8 Binary files /dev/null and b/roc_curves/CvT_ROC_j.png differ diff --git a/roc_curves/CvT_ROC_k.png b/roc_curves/CvT_ROC_k.png new file mode 100644 index 0000000000000000000000000000000000000000..8608910ecbd7c480a790709eea278a7a6967cac5 Binary files /dev/null and b/roc_curves/CvT_ROC_k.png differ diff --git a/roc_curves/CvT_ROC_l.png b/roc_curves/CvT_ROC_l.png new file mode 100644 index 0000000000000000000000000000000000000000..0c99e514f0b6dbe9d80ab69453ae18b83e25565a Binary files /dev/null and b/roc_curves/CvT_ROC_l.png differ diff --git a/training_curves/CvT_accuracy.png b/training_curves/CvT_accuracy.png new file mode 100644 index 0000000000000000000000000000000000000000..e02b3549cf96df89c9194a298bd3eb7f0a38864a Binary files /dev/null and b/training_curves/CvT_accuracy.png differ diff --git a/training_curves/CvT_auc.png b/training_curves/CvT_auc.png new file mode 100644 index 0000000000000000000000000000000000000000..3c859368ba51a81df191ac7ec09e865b55798536 Binary files /dev/null and b/training_curves/CvT_auc.png differ diff --git a/training_curves/CvT_combined_metrics.png b/training_curves/CvT_combined_metrics.png new file mode 100644 index 0000000000000000000000000000000000000000..b8b5e49b9e6b4228a7fbde6dc7f7ad1593e3f8f6 --- /dev/null +++ b/training_curves/CvT_combined_metrics.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f9cddad3d9a2305e35dabb091fd81d5f3f50bad7d285fdf097de841427120b40 +size 166725 diff --git a/training_curves/CvT_f1.png b/training_curves/CvT_f1.png new file mode 100644 index 0000000000000000000000000000000000000000..99426f44b8078c59bf185c4fe0600afa04035fcf Binary files /dev/null and b/training_curves/CvT_f1.png differ diff --git a/training_curves/CvT_loss.png b/training_curves/CvT_loss.png new file mode 100644 index 0000000000000000000000000000000000000000..bc77b53af235525e37cec98f17faafb6c5e8467e Binary files /dev/null and b/training_curves/CvT_loss.png differ diff --git a/training_curves/CvT_metrics.csv b/training_curves/CvT_metrics.csv new file mode 100644 index 0000000000000000000000000000000000000000..11c3d31bc6d14c014b2d77bc2b9cb82d383b7dd3 --- /dev/null +++ b/training_curves/CvT_metrics.csv @@ -0,0 +1,47 @@ +epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1 +1,0.5454545855617974,0.41878439229709413,0.6942059718849403,0.793002915451895,0.7856951154682903,0.8944710537276137,0.6973220935743788,0.8035961272475796 +2,0.4716189381318856,0.37400962720807024,0.7424923897800731,0.8294460641399417,0.846225792479242,0.9113454003008951,0.7393520802499589,0.8294460641399417 +3,0.4498043883640691,0.36516355677526824,0.7571399254369463,0.8250728862973761,0.8614838160290162,0.9176437963773598,0.7559042937192754,0.8288159771754636 +4,0.43645898957468926,0.3376608351055457,0.7647073913192188,0.8534985422740525,0.870118987584132,0.928606490492907,0.7661530878466232,0.85360524399126 +5,0.4270626576820909,0.33928940812283287,0.7684184423846496,0.8411078717201166,0.8753239019916431,0.9327937339033906,0.7710631535346876,0.8481894150417827 +6,0.4177690687775918,0.3458701694498257,0.7748486506823545,0.8483965014577259,0.8810562526193122,0.9344427066953395,0.7753270134901064,0.8567493112947658 +7,0.41739760828490347,0.3428083369752756,0.7742329924410849,0.8447521865889213,0.8811244116648936,0.9428214434461831,0.7766942666001336,0.8561782579338285 +8,0.4087680892544022,0.38979807643779163,0.7779782467421418,0.8126822157434402,0.8860292670244734,0.9330072928796674,0.7794660981679506,0.8305866842452209 +9,0.40660411412927855,0.3142256221079618,0.7802613127201833,0.8520408163265306,0.8873070539085166,0.9437426157468402,0.7814610085891658,0.8579426172148356 +10,0.4018868911731959,0.3325835931231607,0.7824759722269726,0.8520408163265306,0.889890965812465,0.9431497505291162,0.7827750215611098,0.8602890571231934 +11,0.4010628289264208,0.2987588767407587,0.7822279987686835,0.869533527696793,0.8902885009233055,0.9464413212181999,0.785134565089007,0.8725978647686833 +12,0.3981729524995175,0.3306502518257664,0.7838954065054554,0.847667638483965,0.8916990875405698,0.9435046196737754,0.783278308965399,0.8569472963723477 +13,0.3931052854340262,0.3015700927976269,0.7860245579231795,0.8651603498542274,0.8943008290085421,0.9444746661680081,0.7864153906556733,0.8666186012977649 +14,0.3904137644843763,0.2880233984560035,0.7882050141943427,0.8709912536443148,0.8961008655325486,0.951043995274078,0.7891730859258629,0.8731182795698925 +15,0.3862014959101255,0.3049151019000451,0.7903256148031603,0.8658892128279884,0.8981321271951521,0.9465645691846084,0.7905959914260583,0.8698727015558698 +16,0.3882054155839306,0.3170544022895157,0.7883247255190341,0.8564139941690962,0.8968897856469992,0.9471149351035707,0.7883735840991665,0.8638562543192813 +17,0.3851278321350853,0.28406817198842327,0.7916253377569518,0.8760932944606414,0.8989055089189882,0.9520544161021343,0.7955157627986205,0.8782234957020058 +18,0.38282828246470324,0.3047257037534658,0.7919759209221192,0.8673469387755102,0.8998374235231965,0.9480743567731132,0.7940539076256264,0.8727272727272727 +19,0.3831823689526514,0.3018142296864757,0.7938742004993672,0.8651603498542274,0.900099528175156,0.9501557599299613,0.7941874562437033,0.8714384989576095 +20,0.3803508949026174,0.2997558918882042,0.7936005746143585,0.8673469387755102,0.9014052887832577,0.9500909484993498,0.7942269658323672,0.8714689265536724 +21,0.37627306820393247,0.28655936141055804,0.7946950781543934,0.8753644314868805,0.9031548813255292,0.9528491529889757,0.797055144199885,0.8788093550673282 +22,0.3791275146530498,0.31103754304240816,0.7934039060095085,0.8534985422740525,0.9016801477021054,0.9532358966077061,0.7981857516350788,0.8624229979466119 +23,0.37644750438942753,0.31795066668931665,0.7943701474159456,0.8615160349854227,0.9028640619885934,0.9489806543192038,0.7965378953246358,0.8695054945054945 +24,0.37052251767511996,0.2799088840234384,0.7974911926668263,0.8717201166180758,0.9060147108728265,0.9552376135793758,0.7974548222395169,0.875177304964539 +25,0.36707203877373,0.2850041752436766,0.8003557136505114,0.8826530612244898,0.9082701132862941,0.951667672483404,0.8052840511058478,0.8837545126353791 +26,0.36778277345887245,0.2616846166094955,0.8007233984334918,0.8848396501457726,0.9079216477163619,0.9584622478729101,0.8032353661316605,0.886002886002886 +27,0.36552157687697256,0.264149998306533,0.8028354482334029,0.8848396501457726,0.9094252403564035,0.9586109954185756,0.8056014568509089,0.8868194842406877 +28,0.36577115871370763,0.26402455794220414,0.8010483291719397,0.8892128279883382,0.9086845302272508,0.9578290083213628,0.8036059457588777,0.8904899135446686 +29,0.36358578548951015,0.2691162500541342,0.8014844204261723,0.8855685131195336,0.9094048793586998,0.9570927079703184,0.8067298246782438,0.8874551971326164 +30,0.3637443440624353,0.3012346520194507,0.8024848650682355,0.8651603498542274,0.9096946158427448,0.9518047327219101,0.8085152241132048,0.8710801393728222 +31,0.36424110488199757,0.27704282674080427,0.803262988678729,0.8797376093294461,0.9098558322387691,0.9565019677175326,0.8082091294054883,0.8838845883180858 +32,0.36254328096247684,0.2780654456455576,0.8035964702260834,0.8790087463556852,0.9103594376525266,0.9561332863007762,0.8075185827655847,0.882768361581921 +33,0.3634739719800414,0.28042872528119267,0.8026131272018333,0.8760932944606414,0.9098308456093774,0.9572223308315412,0.8052246110229843,0.8812849162011173 +34,0.3610940596292188,0.2735784603364266,0.8035793686082704,0.8790087463556852,0.9111481958874158,0.9572212683490723,0.8046401265489059,0.882768361581921 +35,0.3633829878393609,0.28090626314152084,0.8042805349386052,0.8753644314868805,0.9104869063276791,0.954181506005151,0.8048329197895616,0.8789808917197452 +36,0.3625948407028851,0.29401117797843224,0.8001675958545679,0.8760932944606414,0.9095737659505166,0.9544566889646321,0.8021403051289432,0.8816155988857939 +37,0.36652841117079754,0.28202002423845296,0.8010397783630332,0.8790087463556852,0.9084846772598678,0.9563064709432294,0.8041743814172698,0.8830985915492958 +38,0.36131294218203874,0.2753060121230412,0.8036135718438965,0.8797376093294461,0.9110813176937551,0.9563139083205127,0.8076030593182713,0.8822269807280514 +39,0.3631269225699266,0.29128056512629674,0.8038615453021856,0.8753644314868805,0.9104374160753913,0.9542070055844079,0.8048162014976175,0.8806699232379623 +40,0.36260165742986744,0.27745224704895355,0.8035109621370181,0.880466472303207,0.9103856698917665,0.9557199806203198,0.8046983231202033,0.8836879432624114 +41,0.365013342035874,0.2783045386433949,0.8008773129938093,0.8775510204081632,0.9087424374855558,0.9552854252904827,0.802844685264361,0.8801711840228246 +42,0.36325351914849174,0.2887538745347682,0.8024848650682355,0.8746355685131195,0.9100745935285524,0.9538043247286421,0.8047405303510596,0.8788732394366198 +43,0.36252303772709515,0.2807743189633761,0.8017067414577419,0.8753644314868805,0.9098011751476967,0.9547541840559632,0.8035977437877941,0.8786373314407381 +44,0.3638359864524954,0.2731195484062673,0.8016383349864897,0.8833819241982507,0.9094158889236511,0.9554660473102196,0.8033167720821393,0.8848920863309353 +45,0.36243859979491244,0.27358127413616234,0.8031774805896638,0.8811953352769679,0.9104034476894268,0.9562862837763176,0.805581363920469,0.8836545324768023 +46,0.3623487997137416,0.3242336612401134,0.8035879194171769,0.8513119533527697,0.9104723615880296,0.9527120927504696,0.8057472430823355,0.8623481781376519 diff --git a/training_metrics.csv b/training_metrics.csv new file mode 100644 index 0000000000000000000000000000000000000000..11c3d31bc6d14c014b2d77bc2b9cb82d383b7dd3 --- /dev/null +++ b/training_metrics.csv @@ -0,0 +1,47 @@ +epoch,train_loss,val_loss,train_accuracy,val_accuracy,train_auc,val_auc,train_f1,val_f1 +1,0.5454545855617974,0.41878439229709413,0.6942059718849403,0.793002915451895,0.7856951154682903,0.8944710537276137,0.6973220935743788,0.8035961272475796 +2,0.4716189381318856,0.37400962720807024,0.7424923897800731,0.8294460641399417,0.846225792479242,0.9113454003008951,0.7393520802499589,0.8294460641399417 +3,0.4498043883640691,0.36516355677526824,0.7571399254369463,0.8250728862973761,0.8614838160290162,0.9176437963773598,0.7559042937192754,0.8288159771754636 +4,0.43645898957468926,0.3376608351055457,0.7647073913192188,0.8534985422740525,0.870118987584132,0.928606490492907,0.7661530878466232,0.85360524399126 +5,0.4270626576820909,0.33928940812283287,0.7684184423846496,0.8411078717201166,0.8753239019916431,0.9327937339033906,0.7710631535346876,0.8481894150417827 +6,0.4177690687775918,0.3458701694498257,0.7748486506823545,0.8483965014577259,0.8810562526193122,0.9344427066953395,0.7753270134901064,0.8567493112947658 +7,0.41739760828490347,0.3428083369752756,0.7742329924410849,0.8447521865889213,0.8811244116648936,0.9428214434461831,0.7766942666001336,0.8561782579338285 +8,0.4087680892544022,0.38979807643779163,0.7779782467421418,0.8126822157434402,0.8860292670244734,0.9330072928796674,0.7794660981679506,0.8305866842452209 +9,0.40660411412927855,0.3142256221079618,0.7802613127201833,0.8520408163265306,0.8873070539085166,0.9437426157468402,0.7814610085891658,0.8579426172148356 +10,0.4018868911731959,0.3325835931231607,0.7824759722269726,0.8520408163265306,0.889890965812465,0.9431497505291162,0.7827750215611098,0.8602890571231934 +11,0.4010628289264208,0.2987588767407587,0.7822279987686835,0.869533527696793,0.8902885009233055,0.9464413212181999,0.785134565089007,0.8725978647686833 +12,0.3981729524995175,0.3306502518257664,0.7838954065054554,0.847667638483965,0.8916990875405698,0.9435046196737754,0.783278308965399,0.8569472963723477 +13,0.3931052854340262,0.3015700927976269,0.7860245579231795,0.8651603498542274,0.8943008290085421,0.9444746661680081,0.7864153906556733,0.8666186012977649 +14,0.3904137644843763,0.2880233984560035,0.7882050141943427,0.8709912536443148,0.8961008655325486,0.951043995274078,0.7891730859258629,0.8731182795698925 +15,0.3862014959101255,0.3049151019000451,0.7903256148031603,0.8658892128279884,0.8981321271951521,0.9465645691846084,0.7905959914260583,0.8698727015558698 +16,0.3882054155839306,0.3170544022895157,0.7883247255190341,0.8564139941690962,0.8968897856469992,0.9471149351035707,0.7883735840991665,0.8638562543192813 +17,0.3851278321350853,0.28406817198842327,0.7916253377569518,0.8760932944606414,0.8989055089189882,0.9520544161021343,0.7955157627986205,0.8782234957020058 +18,0.38282828246470324,0.3047257037534658,0.7919759209221192,0.8673469387755102,0.8998374235231965,0.9480743567731132,0.7940539076256264,0.8727272727272727 +19,0.3831823689526514,0.3018142296864757,0.7938742004993672,0.8651603498542274,0.900099528175156,0.9501557599299613,0.7941874562437033,0.8714384989576095 +20,0.3803508949026174,0.2997558918882042,0.7936005746143585,0.8673469387755102,0.9014052887832577,0.9500909484993498,0.7942269658323672,0.8714689265536724 +21,0.37627306820393247,0.28655936141055804,0.7946950781543934,0.8753644314868805,0.9031548813255292,0.9528491529889757,0.797055144199885,0.8788093550673282 +22,0.3791275146530498,0.31103754304240816,0.7934039060095085,0.8534985422740525,0.9016801477021054,0.9532358966077061,0.7981857516350788,0.8624229979466119 +23,0.37644750438942753,0.31795066668931665,0.7943701474159456,0.8615160349854227,0.9028640619885934,0.9489806543192038,0.7965378953246358,0.8695054945054945 +24,0.37052251767511996,0.2799088840234384,0.7974911926668263,0.8717201166180758,0.9060147108728265,0.9552376135793758,0.7974548222395169,0.875177304964539 +25,0.36707203877373,0.2850041752436766,0.8003557136505114,0.8826530612244898,0.9082701132862941,0.951667672483404,0.8052840511058478,0.8837545126353791 +26,0.36778277345887245,0.2616846166094955,0.8007233984334918,0.8848396501457726,0.9079216477163619,0.9584622478729101,0.8032353661316605,0.886002886002886 +27,0.36552157687697256,0.264149998306533,0.8028354482334029,0.8848396501457726,0.9094252403564035,0.9586109954185756,0.8056014568509089,0.8868194842406877 +28,0.36577115871370763,0.26402455794220414,0.8010483291719397,0.8892128279883382,0.9086845302272508,0.9578290083213628,0.8036059457588777,0.8904899135446686 +29,0.36358578548951015,0.2691162500541342,0.8014844204261723,0.8855685131195336,0.9094048793586998,0.9570927079703184,0.8067298246782438,0.8874551971326164 +30,0.3637443440624353,0.3012346520194507,0.8024848650682355,0.8651603498542274,0.9096946158427448,0.9518047327219101,0.8085152241132048,0.8710801393728222 +31,0.36424110488199757,0.27704282674080427,0.803262988678729,0.8797376093294461,0.9098558322387691,0.9565019677175326,0.8082091294054883,0.8838845883180858 +32,0.36254328096247684,0.2780654456455576,0.8035964702260834,0.8790087463556852,0.9103594376525266,0.9561332863007762,0.8075185827655847,0.882768361581921 +33,0.3634739719800414,0.28042872528119267,0.8026131272018333,0.8760932944606414,0.9098308456093774,0.9572223308315412,0.8052246110229843,0.8812849162011173 +34,0.3610940596292188,0.2735784603364266,0.8035793686082704,0.8790087463556852,0.9111481958874158,0.9572212683490723,0.8046401265489059,0.882768361581921 +35,0.3633829878393609,0.28090626314152084,0.8042805349386052,0.8753644314868805,0.9104869063276791,0.954181506005151,0.8048329197895616,0.8789808917197452 +36,0.3625948407028851,0.29401117797843224,0.8001675958545679,0.8760932944606414,0.9095737659505166,0.9544566889646321,0.8021403051289432,0.8816155988857939 +37,0.36652841117079754,0.28202002423845296,0.8010397783630332,0.8790087463556852,0.9084846772598678,0.9563064709432294,0.8041743814172698,0.8830985915492958 +38,0.36131294218203874,0.2753060121230412,0.8036135718438965,0.8797376093294461,0.9110813176937551,0.9563139083205127,0.8076030593182713,0.8822269807280514 +39,0.3631269225699266,0.29128056512629674,0.8038615453021856,0.8753644314868805,0.9104374160753913,0.9542070055844079,0.8048162014976175,0.8806699232379623 +40,0.36260165742986744,0.27745224704895355,0.8035109621370181,0.880466472303207,0.9103856698917665,0.9557199806203198,0.8046983231202033,0.8836879432624114 +41,0.365013342035874,0.2783045386433949,0.8008773129938093,0.8775510204081632,0.9087424374855558,0.9552854252904827,0.802844685264361,0.8801711840228246 +42,0.36325351914849174,0.2887538745347682,0.8024848650682355,0.8746355685131195,0.9100745935285524,0.9538043247286421,0.8047405303510596,0.8788732394366198 +43,0.36252303772709515,0.2807743189633761,0.8017067414577419,0.8753644314868805,0.9098011751476967,0.9547541840559632,0.8035977437877941,0.8786373314407381 +44,0.3638359864524954,0.2731195484062673,0.8016383349864897,0.8833819241982507,0.9094158889236511,0.9554660473102196,0.8033167720821393,0.8848920863309353 +45,0.36243859979491244,0.27358127413616234,0.8031774805896638,0.8811953352769679,0.9104034476894268,0.9562862837763176,0.805581363920469,0.8836545324768023 +46,0.3623487997137416,0.3242336612401134,0.8035879194171769,0.8513119533527697,0.9104723615880296,0.9527120927504696,0.8057472430823355,0.8623481781376519 diff --git a/training_notebook_c2.ipynb b/training_notebook_c2.ipynb new file mode 100644 index 0000000000000000000000000000000000000000..0f8f032a2e59d06139815e5d9ae0c047d9139fdd --- /dev/null +++ b/training_notebook_c2.ipynb @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3840ec461cf0a549c6e39196be842adfbe3b10df2d6dbc623fd1d1a0a8689940 +size 25453968