├── .gitignore ├── LICENSE ├── README.md ├── assets └── figure_1.jpg ├── checkpoints ├── models │ ├── ndt1 │ │ └── placehold │ └── ndt2 │ │ └── placehold └── placehold ├── data ├── 2023_12_bwm_release.csv ├── ibl_100_eids.txt ├── ibl_repro_ephys_eids.txt ├── target_eids.txt ├── test_re_eids.txt └── trials_mask │ ├── 03cf52f6-fba6-4743-a42e-dd1ac3072343.npy │ ├── 03d9a098-07bf-4765-88b7-85f8d8f620cc.npy │ ├── 0802ced5-33a3-405e-8336-b65ebc5cb07c.npy │ ├── 0cad7ea8-8e6c-4ad1-a5c5-53fbb2df1a63.npy │ ├── 15763234-d21e-491f-a01b-1238eb96d389.npy │ ├── 30c4e2ab-dffc-499d-aae4-e51d6b3218c2.npy │ ├── 3638d102-e8b6-4230-8742-e548cd87a949.npy │ ├── 41872d7f-75cb-4445-bb1a-132b354c44f0.npy │ ├── 4b00df29-3769-43be-bb40-128b1cba6d35.npy │ ├── 4b7fbad4-f6de-43b4-9b15-c7c7ef44db4b.npy │ ├── 51e53aff-1d5d-4182-a684-aba783d50ae5.npy │ ├── 572a95d1-39ca-42e1-8424-5c9ffcb2df87.npy │ ├── 57fd2325-67f4-4d45-9907-29e77d3043d7.npy │ ├── 5dcee0eb-b34d-4652-acc3-d10afc6eae68.npy │ ├── 61e11a11-ab65-48fb-ae08-3cb80662e5d6.npy │ ├── 71e55bfe-5a3a-4cba-bdc7-f085140d798e.npy │ ├── 72cb5550-43b4-4ef0-add5-e4adfdfb5e02.npy │ ├── 754b74d5-7a06-4004-ae0c-72a10b6ed2e6.npy │ ├── 7691eeb3-715b-4571-8fda-6bb57aab8253.npy │ ├── 781b35fd-e1f0-4d14-b2bb-95b7263082bb.npy │ ├── 7cb81727-2097-4b52-b480-c89867b5b34c.npy │ ├── 824cf03d-4012-4ab1-b499-c83a92c5589e.npy │ ├── 862ade13-53cd-4221-a3fa-dda8643641f2.npy │ ├── 88224abb-5746-431f-9c17-17d7ef806e6a.npy │ ├── a66f1593-dafd-4982-9b66-f9554b6c86b5.npy │ ├── a8a8af78-16de-4841-ab07-fde4b5281a03.npy │ ├── b03fbc44-3d8e-4a6c-8a50-5ea3498568e0.npy │ ├── c4432264-e1ae-446f-8a07-6280abade813.npy │ ├── c51f34d8-42f6-4c9c-bb5b-669fd9c42cd9.npy │ ├── c7248e09-8c0d-40f2-9eb4-700a8973d8c8.npy │ ├── c7bf2d49-4937-4597-b307-9f39cb1c7b16.npy │ ├── d0ea3148-948d-4817-94f8-dcaf2342bbbe.npy │ ├── d2832a38-27f6-452d-91d6-af72d794136c.npy │ ├── d57df551-6dcb-4242-9c72-b806cff5613a.npy │ ├── d9f0c293-df4c-410a-846d-842e47c6b502.npy │ ├── dac3a4c1-b666-4de0-87e8-8c514483cacf.npy │ ├── db4df448-e449-4a6f-a0e7-288711e7a75a.npy │ ├── ecb5520d-1358-434c-95ec-93687ecd1396.npy │ ├── f140a2ec-fd49-4814-994a-fe3476f14e66.npy │ └── f312aaec-3b6f-44b3-86b4-3a0c119c0438.npy ├── env.yaml ├── notebooks ├── data_loading_examples.ipynb └── model_evaluation.ipynb ├── script ├── draw.sh ├── finetune_eval_multi_session.sh ├── run_finetune_multi_session.sh └── train_sessions.sh └── src ├── behavior_decoders ├── decoder_loader.py ├── eval.py ├── hyperparam_tuning.py └── models.py ├── configs ├── co_smooth_ndt1.yaml ├── decoder.yaml ├── decoder_trainer.yaml ├── finetune_sessions_trainer.yaml ├── itransformer.yaml ├── ndt1.yaml ├── ndt1_causal.yaml ├── ndt1_causal_eval.yaml ├── ndt1_eval.yaml ├── ndt1_mask_token.yaml ├── ndt1_mask_token_eval.yaml ├── ndt1_prompting.yaml ├── ndt1_prompting_eval.yaml ├── ndt1_stitching.yaml ├── ndt1_stitching_eval.yaml ├── ndt1_stitching_prompting.yaml ├── ndt1_stitching_prompting_eval.yaml ├── ndt2.yaml ├── ndt2_causal.yaml ├── ndt2_causal_eval.yaml ├── ndt2_eval.yaml ├── ndt2_prompting.yaml ├── ndt2_prompting_eval.yaml ├── ppwang │ ├── trainer_sl_choice_ndt1.yaml │ └── trainer_sl_continuous_ndt1.yaml ├── ssl_session_trainer.yaml ├── ssl_sessions_trainer.yaml ├── stpatch.yaml ├── sweep.yaml ├── trainer_ndt1.yaml ├── trainer_ndt2.yaml ├── trainer_sl_choice_ndt1.yaml ├── trainer_sl_choice_ndt2.yaml ├── trainer_sl_continuous_ndt1.yaml └── trainer_sl_continuous_ndt2.yaml ├── data_loading_example.py ├── decode_multi_session.py ├── decode_single_session.py ├── draw_table.py ├── finetune_eval_multi_session.py ├── loader ├── base.py └── make_loader.py ├── models ├── __init__.py ├── itransformer.py ├── masker.py ├── model_output.py ├── ndt1.py ├── patcher.py ├── region_lookup.py └── stpatch.py ├── setup.py ├── train_sessions.py ├── trainer ├── __init__.py ├── base.py └── make.py └── utils ├── __init__.py ├── config_utils.py ├── dataset_utils.py ├── draw_ndt2.py ├── eval_utils.py ├── ibl_data_utils.py ├── metric_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/README.md -------------------------------------------------------------------------------- /assets/figure_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/assets/figure_1.jpg -------------------------------------------------------------------------------- /checkpoints/models/ndt1/placehold: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /checkpoints/models/ndt2/placehold: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /checkpoints/placehold: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/2023_12_bwm_release.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/2023_12_bwm_release.csv -------------------------------------------------------------------------------- /data/ibl_100_eids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/ibl_100_eids.txt -------------------------------------------------------------------------------- /data/ibl_repro_ephys_eids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/ibl_repro_ephys_eids.txt -------------------------------------------------------------------------------- /data/target_eids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/target_eids.txt -------------------------------------------------------------------------------- /data/test_re_eids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/test_re_eids.txt -------------------------------------------------------------------------------- /data/trials_mask/03cf52f6-fba6-4743-a42e-dd1ac3072343.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/03cf52f6-fba6-4743-a42e-dd1ac3072343.npy -------------------------------------------------------------------------------- /data/trials_mask/03d9a098-07bf-4765-88b7-85f8d8f620cc.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/03d9a098-07bf-4765-88b7-85f8d8f620cc.npy -------------------------------------------------------------------------------- /data/trials_mask/0802ced5-33a3-405e-8336-b65ebc5cb07c.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/0802ced5-33a3-405e-8336-b65ebc5cb07c.npy -------------------------------------------------------------------------------- /data/trials_mask/0cad7ea8-8e6c-4ad1-a5c5-53fbb2df1a63.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/0cad7ea8-8e6c-4ad1-a5c5-53fbb2df1a63.npy -------------------------------------------------------------------------------- /data/trials_mask/15763234-d21e-491f-a01b-1238eb96d389.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/15763234-d21e-491f-a01b-1238eb96d389.npy -------------------------------------------------------------------------------- /data/trials_mask/30c4e2ab-dffc-499d-aae4-e51d6b3218c2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/30c4e2ab-dffc-499d-aae4-e51d6b3218c2.npy -------------------------------------------------------------------------------- /data/trials_mask/3638d102-e8b6-4230-8742-e548cd87a949.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/3638d102-e8b6-4230-8742-e548cd87a949.npy -------------------------------------------------------------------------------- /data/trials_mask/41872d7f-75cb-4445-bb1a-132b354c44f0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/41872d7f-75cb-4445-bb1a-132b354c44f0.npy -------------------------------------------------------------------------------- /data/trials_mask/4b00df29-3769-43be-bb40-128b1cba6d35.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/4b00df29-3769-43be-bb40-128b1cba6d35.npy -------------------------------------------------------------------------------- /data/trials_mask/4b7fbad4-f6de-43b4-9b15-c7c7ef44db4b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/4b7fbad4-f6de-43b4-9b15-c7c7ef44db4b.npy -------------------------------------------------------------------------------- /data/trials_mask/51e53aff-1d5d-4182-a684-aba783d50ae5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/51e53aff-1d5d-4182-a684-aba783d50ae5.npy -------------------------------------------------------------------------------- /data/trials_mask/572a95d1-39ca-42e1-8424-5c9ffcb2df87.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/572a95d1-39ca-42e1-8424-5c9ffcb2df87.npy -------------------------------------------------------------------------------- /data/trials_mask/57fd2325-67f4-4d45-9907-29e77d3043d7.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/57fd2325-67f4-4d45-9907-29e77d3043d7.npy -------------------------------------------------------------------------------- /data/trials_mask/5dcee0eb-b34d-4652-acc3-d10afc6eae68.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/5dcee0eb-b34d-4652-acc3-d10afc6eae68.npy -------------------------------------------------------------------------------- /data/trials_mask/61e11a11-ab65-48fb-ae08-3cb80662e5d6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/61e11a11-ab65-48fb-ae08-3cb80662e5d6.npy -------------------------------------------------------------------------------- /data/trials_mask/71e55bfe-5a3a-4cba-bdc7-f085140d798e.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/71e55bfe-5a3a-4cba-bdc7-f085140d798e.npy -------------------------------------------------------------------------------- /data/trials_mask/72cb5550-43b4-4ef0-add5-e4adfdfb5e02.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/72cb5550-43b4-4ef0-add5-e4adfdfb5e02.npy -------------------------------------------------------------------------------- /data/trials_mask/754b74d5-7a06-4004-ae0c-72a10b6ed2e6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/754b74d5-7a06-4004-ae0c-72a10b6ed2e6.npy -------------------------------------------------------------------------------- /data/trials_mask/7691eeb3-715b-4571-8fda-6bb57aab8253.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/7691eeb3-715b-4571-8fda-6bb57aab8253.npy -------------------------------------------------------------------------------- /data/trials_mask/781b35fd-e1f0-4d14-b2bb-95b7263082bb.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/781b35fd-e1f0-4d14-b2bb-95b7263082bb.npy -------------------------------------------------------------------------------- /data/trials_mask/7cb81727-2097-4b52-b480-c89867b5b34c.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/7cb81727-2097-4b52-b480-c89867b5b34c.npy -------------------------------------------------------------------------------- /data/trials_mask/824cf03d-4012-4ab1-b499-c83a92c5589e.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/824cf03d-4012-4ab1-b499-c83a92c5589e.npy -------------------------------------------------------------------------------- /data/trials_mask/862ade13-53cd-4221-a3fa-dda8643641f2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/862ade13-53cd-4221-a3fa-dda8643641f2.npy -------------------------------------------------------------------------------- /data/trials_mask/88224abb-5746-431f-9c17-17d7ef806e6a.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/88224abb-5746-431f-9c17-17d7ef806e6a.npy -------------------------------------------------------------------------------- /data/trials_mask/a66f1593-dafd-4982-9b66-f9554b6c86b5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/a66f1593-dafd-4982-9b66-f9554b6c86b5.npy -------------------------------------------------------------------------------- /data/trials_mask/a8a8af78-16de-4841-ab07-fde4b5281a03.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/a8a8af78-16de-4841-ab07-fde4b5281a03.npy -------------------------------------------------------------------------------- /data/trials_mask/b03fbc44-3d8e-4a6c-8a50-5ea3498568e0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/b03fbc44-3d8e-4a6c-8a50-5ea3498568e0.npy -------------------------------------------------------------------------------- /data/trials_mask/c4432264-e1ae-446f-8a07-6280abade813.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/c4432264-e1ae-446f-8a07-6280abade813.npy -------------------------------------------------------------------------------- /data/trials_mask/c51f34d8-42f6-4c9c-bb5b-669fd9c42cd9.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/c51f34d8-42f6-4c9c-bb5b-669fd9c42cd9.npy -------------------------------------------------------------------------------- /data/trials_mask/c7248e09-8c0d-40f2-9eb4-700a8973d8c8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/c7248e09-8c0d-40f2-9eb4-700a8973d8c8.npy -------------------------------------------------------------------------------- /data/trials_mask/c7bf2d49-4937-4597-b307-9f39cb1c7b16.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/c7bf2d49-4937-4597-b307-9f39cb1c7b16.npy -------------------------------------------------------------------------------- /data/trials_mask/d0ea3148-948d-4817-94f8-dcaf2342bbbe.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/d0ea3148-948d-4817-94f8-dcaf2342bbbe.npy -------------------------------------------------------------------------------- /data/trials_mask/d2832a38-27f6-452d-91d6-af72d794136c.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/d2832a38-27f6-452d-91d6-af72d794136c.npy -------------------------------------------------------------------------------- /data/trials_mask/d57df551-6dcb-4242-9c72-b806cff5613a.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/d57df551-6dcb-4242-9c72-b806cff5613a.npy -------------------------------------------------------------------------------- /data/trials_mask/d9f0c293-df4c-410a-846d-842e47c6b502.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/d9f0c293-df4c-410a-846d-842e47c6b502.npy -------------------------------------------------------------------------------- /data/trials_mask/dac3a4c1-b666-4de0-87e8-8c514483cacf.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/dac3a4c1-b666-4de0-87e8-8c514483cacf.npy -------------------------------------------------------------------------------- /data/trials_mask/db4df448-e449-4a6f-a0e7-288711e7a75a.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/db4df448-e449-4a6f-a0e7-288711e7a75a.npy -------------------------------------------------------------------------------- /data/trials_mask/ecb5520d-1358-434c-95ec-93687ecd1396.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/ecb5520d-1358-434c-95ec-93687ecd1396.npy -------------------------------------------------------------------------------- /data/trials_mask/f140a2ec-fd49-4814-994a-fe3476f14e66.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/f140a2ec-fd49-4814-994a-fe3476f14e66.npy -------------------------------------------------------------------------------- /data/trials_mask/f312aaec-3b6f-44b3-86b4-3a0c119c0438.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/data/trials_mask/f312aaec-3b6f-44b3-86b4-3a0c119c0438.npy -------------------------------------------------------------------------------- /env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/env.yaml -------------------------------------------------------------------------------- /notebooks/data_loading_examples.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/notebooks/data_loading_examples.ipynb -------------------------------------------------------------------------------- /notebooks/model_evaluation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/notebooks/model_evaluation.ipynb -------------------------------------------------------------------------------- /script/draw.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/script/draw.sh -------------------------------------------------------------------------------- /script/finetune_eval_multi_session.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/script/finetune_eval_multi_session.sh -------------------------------------------------------------------------------- /script/run_finetune_multi_session.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/script/run_finetune_multi_session.sh -------------------------------------------------------------------------------- /script/train_sessions.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/script/train_sessions.sh -------------------------------------------------------------------------------- /src/behavior_decoders/decoder_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/behavior_decoders/decoder_loader.py -------------------------------------------------------------------------------- /src/behavior_decoders/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/behavior_decoders/eval.py -------------------------------------------------------------------------------- /src/behavior_decoders/hyperparam_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/behavior_decoders/hyperparam_tuning.py -------------------------------------------------------------------------------- /src/behavior_decoders/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/behavior_decoders/models.py -------------------------------------------------------------------------------- /src/configs/co_smooth_ndt1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/co_smooth_ndt1.yaml -------------------------------------------------------------------------------- /src/configs/decoder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/decoder.yaml -------------------------------------------------------------------------------- /src/configs/decoder_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/decoder_trainer.yaml -------------------------------------------------------------------------------- /src/configs/finetune_sessions_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/finetune_sessions_trainer.yaml -------------------------------------------------------------------------------- /src/configs/itransformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/itransformer.yaml -------------------------------------------------------------------------------- /src/configs/ndt1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_causal.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_causal_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_causal_eval.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_eval.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_mask_token.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_mask_token.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_mask_token_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_mask_token_eval.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_prompting.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_prompting.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_prompting_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_prompting_eval.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_stitching.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_stitching.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_stitching_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_stitching_eval.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_stitching_prompting.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_stitching_prompting.yaml -------------------------------------------------------------------------------- /src/configs/ndt1_stitching_prompting_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt1_stitching_prompting_eval.yaml -------------------------------------------------------------------------------- /src/configs/ndt2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt2.yaml -------------------------------------------------------------------------------- /src/configs/ndt2_causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt2_causal.yaml -------------------------------------------------------------------------------- /src/configs/ndt2_causal_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt2_causal_eval.yaml -------------------------------------------------------------------------------- /src/configs/ndt2_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt2_eval.yaml -------------------------------------------------------------------------------- /src/configs/ndt2_prompting.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt2_prompting.yaml -------------------------------------------------------------------------------- /src/configs/ndt2_prompting_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ndt2_prompting_eval.yaml -------------------------------------------------------------------------------- /src/configs/ppwang/trainer_sl_choice_ndt1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ppwang/trainer_sl_choice_ndt1.yaml -------------------------------------------------------------------------------- /src/configs/ppwang/trainer_sl_continuous_ndt1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ppwang/trainer_sl_continuous_ndt1.yaml -------------------------------------------------------------------------------- /src/configs/ssl_session_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ssl_session_trainer.yaml -------------------------------------------------------------------------------- /src/configs/ssl_sessions_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/ssl_sessions_trainer.yaml -------------------------------------------------------------------------------- /src/configs/stpatch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/stpatch.yaml -------------------------------------------------------------------------------- /src/configs/sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/sweep.yaml -------------------------------------------------------------------------------- /src/configs/trainer_ndt1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/trainer_ndt1.yaml -------------------------------------------------------------------------------- /src/configs/trainer_ndt2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/trainer_ndt2.yaml -------------------------------------------------------------------------------- /src/configs/trainer_sl_choice_ndt1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/trainer_sl_choice_ndt1.yaml -------------------------------------------------------------------------------- /src/configs/trainer_sl_choice_ndt2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/trainer_sl_choice_ndt2.yaml -------------------------------------------------------------------------------- /src/configs/trainer_sl_continuous_ndt1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/trainer_sl_continuous_ndt1.yaml -------------------------------------------------------------------------------- /src/configs/trainer_sl_continuous_ndt2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/configs/trainer_sl_continuous_ndt2.yaml -------------------------------------------------------------------------------- /src/data_loading_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/data_loading_example.py -------------------------------------------------------------------------------- /src/decode_multi_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/decode_multi_session.py -------------------------------------------------------------------------------- /src/decode_single_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/decode_single_session.py -------------------------------------------------------------------------------- /src/draw_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/draw_table.py -------------------------------------------------------------------------------- /src/finetune_eval_multi_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/finetune_eval_multi_session.py -------------------------------------------------------------------------------- /src/loader/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/loader/base.py -------------------------------------------------------------------------------- /src/loader/make_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/loader/make_loader.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/itransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/models/itransformer.py -------------------------------------------------------------------------------- /src/models/masker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/models/masker.py -------------------------------------------------------------------------------- /src/models/model_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/models/model_output.py -------------------------------------------------------------------------------- /src/models/ndt1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/models/ndt1.py -------------------------------------------------------------------------------- /src/models/patcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/models/patcher.py -------------------------------------------------------------------------------- /src/models/region_lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/models/region_lookup.py -------------------------------------------------------------------------------- /src/models/stpatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/models/stpatch.py -------------------------------------------------------------------------------- /src/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/setup.py -------------------------------------------------------------------------------- /src/train_sessions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/train_sessions.py -------------------------------------------------------------------------------- /src/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/trainer/base.py -------------------------------------------------------------------------------- /src/trainer/make.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/trainer/make.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/utils/config_utils.py -------------------------------------------------------------------------------- /src/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/utils/dataset_utils.py -------------------------------------------------------------------------------- /src/utils/draw_ndt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/utils/draw_ndt2.py -------------------------------------------------------------------------------- /src/utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/utils/eval_utils.py -------------------------------------------------------------------------------- /src/utils/ibl_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/utils/ibl_data_utils.py -------------------------------------------------------------------------------- /src/utils/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/utils/metric_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/colehurwitz/IBL_MtM_model/HEAD/src/utils/utils.py --------------------------------------------------------------------------------