├── .gitignore ├── EDA ├── EDA_pointselecter.ipynb ├── eda.ipynb ├── eda_statistics.ipynb └── huggingfacedataset_creating.ipynb ├── Iafoss ├── fastai_fix.py ├── init5.ipynb ├── init5_dynedge.ipynb ├── init5_hb.ipynb ├── init5_hb_ft-2.ipynb ├── init5_hb_ft.ipynb └── loss_functions.py ├── README.md ├── config.py ├── eval.py ├── hb_training_loop ├── BASELINE.ipynb ├── BASELINEFT.ipynb ├── BASELINEFT2.ipynb ├── V17.ipynb ├── V18.ipynb ├── V18FT.ipynb ├── V18FT2-64.ipynb ├── V18FT2.ipynb ├── V19.ipynb ├── V19FT.ipynb ├── V20.ipynb ├── V20FT.ipynb ├── V20FT2.ipynb ├── V20FT2 │ └── Screenshot 2023-03-31 at 10.56.19 AM.png ├── V20FT3.ipynb ├── V22.ipynb ├── V22FT.ipynb ├── V22FT2.ipynb ├── V22FT3.ipynb ├── V22FT4.ipynb ├── V22FT5.ipynb ├── V22FT6.ipynb ├── V23.ipynb ├── V23FT.ipynb ├── V23FT2.ipynb ├── V23FT3.ipynb ├── V23FT4.ipynb ├── V23FT5.ipynb ├── VFTV3_4REL.ipynb ├── VFTV3_4RELFT.ipynb ├── baselineV3_BE_globalrel_d32_0.ipynb ├── baselineV3_BE_globalrel_d64_0_3emaFT.ipynb ├── fastai_fix.py ├── hb_egnn.ipynb ├── hb_global_graphlocal.ipynb ├── hb_graph_encoder.ipynb ├── hb_localattention-ft-2.ipynb ├── hb_localattention-ft.ipynb ├── hb_localattention.ipynb ├── hb_localattentionV2.ipynb ├── hb_localattentionV3.ipynb ├── hb_localattentionV4.ipynb ├── hb_localattentionV4FT.ipynb ├── hb_localattentionV4FT2.ipynb ├── hb_localattentionV4FT2SWA.ipynb ├── hb_mat-v2.ipynb ├── hb_mat.ipynb └── loss_functions.py ├── icecube ├── __init__.py ├── _modidx.py ├── data_test.py ├── data_train.py ├── data_train_v3.py ├── dataset.py ├── fastai_fix.py ├── graphdataset.py ├── loss.py ├── loss_functions.py ├── maximdataset.py ├── model_transformer_base.py ├── models.py ├── modelsgraph.py ├── modelsgraph_k.py ├── modelsgraphk.py └── utils.py ├── nbs ├── 00_dataset.ipynb ├── 00_graphdataset.ipynb ├── 00_maximdataset.ipynb ├── 00_utils.ipynb ├── 01_models.ipynb ├── 02_modelsgraph.ipynb ├── _quarto.yml ├── nbdev.yml └── sidebar.yml ├── settings.ini ├── setup.py ├── train.py └── train_accel.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/.gitignore -------------------------------------------------------------------------------- /EDA/EDA_pointselecter.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/EDA/EDA_pointselecter.ipynb -------------------------------------------------------------------------------- /EDA/eda.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/EDA/eda.ipynb -------------------------------------------------------------------------------- /EDA/eda_statistics.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/EDA/eda_statistics.ipynb -------------------------------------------------------------------------------- /EDA/huggingfacedataset_creating.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/EDA/huggingfacedataset_creating.ipynb -------------------------------------------------------------------------------- /Iafoss/fastai_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/Iafoss/fastai_fix.py -------------------------------------------------------------------------------- /Iafoss/init5.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/Iafoss/init5.ipynb -------------------------------------------------------------------------------- /Iafoss/init5_dynedge.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/Iafoss/init5_dynedge.ipynb -------------------------------------------------------------------------------- /Iafoss/init5_hb.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/Iafoss/init5_hb.ipynb -------------------------------------------------------------------------------- /Iafoss/init5_hb_ft-2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/Iafoss/init5_hb_ft-2.ipynb -------------------------------------------------------------------------------- /Iafoss/init5_hb_ft.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/Iafoss/init5_hb_ft.ipynb -------------------------------------------------------------------------------- /Iafoss/loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/Iafoss/loss_functions.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/config.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/eval.py -------------------------------------------------------------------------------- /hb_training_loop/BASELINE.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/BASELINE.ipynb -------------------------------------------------------------------------------- /hb_training_loop/BASELINEFT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/BASELINEFT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/BASELINEFT2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/BASELINEFT2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V17.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V17.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V18.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V18.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V18FT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V18FT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V18FT2-64.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V18FT2-64.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V18FT2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V18FT2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V19.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V19.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V19FT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V19FT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V20.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V20.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V20FT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V20FT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V20FT2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V20FT2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V20FT2/Screenshot 2023-03-31 at 10.56.19 AM.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V20FT2/Screenshot 2023-03-31 at 10.56.19 AM.png -------------------------------------------------------------------------------- /hb_training_loop/V20FT3.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V20FT3.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V22.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V22.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V22FT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V22FT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V22FT2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V22FT2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V22FT3.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V22FT3.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V22FT4.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V22FT4.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V22FT5.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V22FT5.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V22FT6.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V22FT6.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V23.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V23.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V23FT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V23FT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V23FT2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V23FT2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V23FT3.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V23FT3.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V23FT4.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V23FT4.ipynb -------------------------------------------------------------------------------- /hb_training_loop/V23FT5.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/V23FT5.ipynb -------------------------------------------------------------------------------- /hb_training_loop/VFTV3_4REL.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/VFTV3_4REL.ipynb -------------------------------------------------------------------------------- /hb_training_loop/VFTV3_4RELFT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/VFTV3_4RELFT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/baselineV3_BE_globalrel_d32_0.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/baselineV3_BE_globalrel_d32_0.ipynb -------------------------------------------------------------------------------- /hb_training_loop/baselineV3_BE_globalrel_d64_0_3emaFT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/baselineV3_BE_globalrel_d64_0_3emaFT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/fastai_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/fastai_fix.py -------------------------------------------------------------------------------- /hb_training_loop/hb_egnn.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_egnn.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_global_graphlocal.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_global_graphlocal.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_graph_encoder.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_graph_encoder.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattention-ft-2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattention-ft-2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattention-ft.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattention-ft.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattention.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattention.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattentionV2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattentionV2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattentionV3.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattentionV3.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattentionV4.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattentionV4.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattentionV4FT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattentionV4FT.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattentionV4FT2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattentionV4FT2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_localattentionV4FT2SWA.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_localattentionV4FT2SWA.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_mat-v2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_mat-v2.ipynb -------------------------------------------------------------------------------- /hb_training_loop/hb_mat.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/hb_mat.ipynb -------------------------------------------------------------------------------- /hb_training_loop/loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/hb_training_loop/loss_functions.py -------------------------------------------------------------------------------- /icecube/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.0.1" 2 | -------------------------------------------------------------------------------- /icecube/_modidx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/_modidx.py -------------------------------------------------------------------------------- /icecube/data_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/data_test.py -------------------------------------------------------------------------------- /icecube/data_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/data_train.py -------------------------------------------------------------------------------- /icecube/data_train_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/data_train_v3.py -------------------------------------------------------------------------------- /icecube/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/dataset.py -------------------------------------------------------------------------------- /icecube/fastai_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/fastai_fix.py -------------------------------------------------------------------------------- /icecube/graphdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/graphdataset.py -------------------------------------------------------------------------------- /icecube/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/loss.py -------------------------------------------------------------------------------- /icecube/loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/loss_functions.py -------------------------------------------------------------------------------- /icecube/maximdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/maximdataset.py -------------------------------------------------------------------------------- /icecube/model_transformer_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/model_transformer_base.py -------------------------------------------------------------------------------- /icecube/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/models.py -------------------------------------------------------------------------------- /icecube/modelsgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/modelsgraph.py -------------------------------------------------------------------------------- /icecube/modelsgraph_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/modelsgraph_k.py -------------------------------------------------------------------------------- /icecube/modelsgraphk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/modelsgraphk.py -------------------------------------------------------------------------------- /icecube/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/icecube/utils.py -------------------------------------------------------------------------------- /nbs/00_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/00_dataset.ipynb -------------------------------------------------------------------------------- /nbs/00_graphdataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/00_graphdataset.ipynb -------------------------------------------------------------------------------- /nbs/00_maximdataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/00_maximdataset.ipynb -------------------------------------------------------------------------------- /nbs/00_utils.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/00_utils.ipynb -------------------------------------------------------------------------------- /nbs/01_models.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/01_models.ipynb -------------------------------------------------------------------------------- /nbs/02_modelsgraph.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/02_modelsgraph.ipynb -------------------------------------------------------------------------------- /nbs/_quarto.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/_quarto.yml -------------------------------------------------------------------------------- /nbs/nbdev.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/nbdev.yml -------------------------------------------------------------------------------- /nbs/sidebar.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/nbs/sidebar.yml -------------------------------------------------------------------------------- /settings.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/settings.ini -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/setup.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/train.py -------------------------------------------------------------------------------- /train_accel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DrHB/icecube-journal/HEAD/train_accel.py --------------------------------------------------------------------------------