├── .dockerignore ├── .github └── workflows │ ├── black.yml │ ├── isort.yml │ ├── mypy.yml │ └── test.yml ├── .gitignore ├── .pre-commit-config.yaml ├── Dockerfile ├── LICENSE ├── Makefile ├── README.md ├── docker-compose.yml ├── jupyter ├── jupyter_notebook_config.py └── run_jupyter.sh ├── neuralpredictors ├── __init__.py ├── data │ ├── __init__.py │ ├── datasets │ │ ├── __init__.py │ │ ├── base.py │ │ ├── movies.py │ │ ├── old_datasets.py │ │ └── statics │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ └── filetree.py │ ├── exceptions.py │ ├── samplers.py │ ├── transforms.py │ └── utils.py ├── distributions.py ├── layers │ ├── __init__.py │ ├── activations.py │ ├── affine.py │ ├── attention.py │ ├── conv.py │ ├── cores │ │ ├── __init__.py │ │ ├── base.py │ │ ├── conv2d.py │ │ └── conv3d.py │ ├── encoders │ │ ├── __init__.py │ │ ├── base.py │ │ ├── distribution_encoders.py │ │ ├── encoder3d.py │ │ ├── firing_rate.py │ │ ├── mean_variance_functions.py │ │ └── zero_inflation_encoders.py │ ├── hermite.py │ ├── modulators │ │ ├── __init__.py │ │ └── mlp.py │ ├── perspective │ │ ├── __init__.py │ │ └── mlp.py │ ├── readouts │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── base.py │ │ ├── factorized.py │ │ ├── gaussian.py │ │ ├── multi_readout.py │ │ ├── old_readouts.py │ │ ├── point_pooled.py │ │ └── pyramid.py │ ├── rnn_modules │ │ └── gru_module.py │ ├── shifters │ │ ├── __init__.py │ │ ├── base.py │ │ ├── mlp.py │ │ └── static_affine.py │ └── squeeze_excitation.py ├── measures │ ├── __init__.py │ ├── functions.py │ ├── modules.py │ ├── np_functions.py │ └── zero_inflated_losses.py ├── old_constraints.py ├── regularizers.py ├── training │ ├── __init__.py │ ├── context_managers.py │ ├── cyclers.py │ ├── early_stopping.py │ ├── tracking.py │ └── utils.py └── utils.py ├── notebooks ├── 2dcores_starter.ipynb ├── example_sysident_models.ipynb └── readouts_tutorial.ipynb ├── pyproject.toml ├── requirements-dev.txt ├── setup.py └── test ├── layers └── cores │ └── test_conv2d.py ├── test_TransferLearningCore.py ├── test_constraints.py ├── test_training ├── __init__.py └── test_early_stopping.py └── utils_for_tests.py /.dockerignore: -------------------------------------------------------------------------------- 1 | .mypy_cache/ 2 | -------------------------------------------------------------------------------- /.github/workflows/black.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/.github/workflows/black.yml -------------------------------------------------------------------------------- /.github/workflows/isort.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/.github/workflows/isort.yml -------------------------------------------------------------------------------- /.github/workflows/mypy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/.github/workflows/mypy.yml -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/README.md -------------------------------------------------------------------------------- /docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/docker-compose.yml -------------------------------------------------------------------------------- /jupyter/jupyter_notebook_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/jupyter/jupyter_notebook_config.py -------------------------------------------------------------------------------- /jupyter/run_jupyter.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | jupyter lab "$@" --allow-root 4 | -------------------------------------------------------------------------------- /neuralpredictors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /neuralpredictors/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /neuralpredictors/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/datasets/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/data/datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/datasets/base.py -------------------------------------------------------------------------------- /neuralpredictors/data/datasets/movies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/datasets/movies.py -------------------------------------------------------------------------------- /neuralpredictors/data/datasets/old_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/datasets/old_datasets.py -------------------------------------------------------------------------------- /neuralpredictors/data/datasets/statics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/datasets/statics/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/data/datasets/statics/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/datasets/statics/base.py -------------------------------------------------------------------------------- /neuralpredictors/data/datasets/statics/filetree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/datasets/statics/filetree.py -------------------------------------------------------------------------------- /neuralpredictors/data/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/exceptions.py -------------------------------------------------------------------------------- /neuralpredictors/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/samplers.py -------------------------------------------------------------------------------- /neuralpredictors/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/transforms.py -------------------------------------------------------------------------------- /neuralpredictors/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/data/utils.py -------------------------------------------------------------------------------- /neuralpredictors/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/distributions.py -------------------------------------------------------------------------------- /neuralpredictors/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/activations.py -------------------------------------------------------------------------------- /neuralpredictors/layers/affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/affine.py -------------------------------------------------------------------------------- /neuralpredictors/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/attention.py -------------------------------------------------------------------------------- /neuralpredictors/layers/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/conv.py -------------------------------------------------------------------------------- /neuralpredictors/layers/cores/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/cores/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/layers/cores/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/cores/base.py -------------------------------------------------------------------------------- /neuralpredictors/layers/cores/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/cores/conv2d.py -------------------------------------------------------------------------------- /neuralpredictors/layers/cores/conv3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/cores/conv3d.py -------------------------------------------------------------------------------- /neuralpredictors/layers/encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/encoders/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/layers/encoders/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/encoders/base.py -------------------------------------------------------------------------------- /neuralpredictors/layers/encoders/distribution_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/encoders/distribution_encoders.py -------------------------------------------------------------------------------- /neuralpredictors/layers/encoders/encoder3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/encoders/encoder3d.py -------------------------------------------------------------------------------- /neuralpredictors/layers/encoders/firing_rate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/encoders/firing_rate.py -------------------------------------------------------------------------------- /neuralpredictors/layers/encoders/mean_variance_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/encoders/mean_variance_functions.py -------------------------------------------------------------------------------- /neuralpredictors/layers/encoders/zero_inflation_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/encoders/zero_inflation_encoders.py -------------------------------------------------------------------------------- /neuralpredictors/layers/hermite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/hermite.py -------------------------------------------------------------------------------- /neuralpredictors/layers/modulators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /neuralpredictors/layers/modulators/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/modulators/mlp.py -------------------------------------------------------------------------------- /neuralpredictors/layers/perspective/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /neuralpredictors/layers/perspective/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/perspective/mlp.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/attention.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/base.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/factorized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/factorized.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/gaussian.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/multi_readout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/multi_readout.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/old_readouts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/old_readouts.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/point_pooled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/point_pooled.py -------------------------------------------------------------------------------- /neuralpredictors/layers/readouts/pyramid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/readouts/pyramid.py -------------------------------------------------------------------------------- /neuralpredictors/layers/rnn_modules/gru_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/rnn_modules/gru_module.py -------------------------------------------------------------------------------- /neuralpredictors/layers/shifters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/shifters/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/layers/shifters/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/shifters/base.py -------------------------------------------------------------------------------- /neuralpredictors/layers/shifters/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/shifters/mlp.py -------------------------------------------------------------------------------- /neuralpredictors/layers/shifters/static_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/shifters/static_affine.py -------------------------------------------------------------------------------- /neuralpredictors/layers/squeeze_excitation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/layers/squeeze_excitation.py -------------------------------------------------------------------------------- /neuralpredictors/measures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/measures/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/measures/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/measures/functions.py -------------------------------------------------------------------------------- /neuralpredictors/measures/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/measures/modules.py -------------------------------------------------------------------------------- /neuralpredictors/measures/np_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/measures/np_functions.py -------------------------------------------------------------------------------- /neuralpredictors/measures/zero_inflated_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/measures/zero_inflated_losses.py -------------------------------------------------------------------------------- /neuralpredictors/old_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/old_constraints.py -------------------------------------------------------------------------------- /neuralpredictors/regularizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/regularizers.py -------------------------------------------------------------------------------- /neuralpredictors/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/training/__init__.py -------------------------------------------------------------------------------- /neuralpredictors/training/context_managers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/training/context_managers.py -------------------------------------------------------------------------------- /neuralpredictors/training/cyclers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/training/cyclers.py -------------------------------------------------------------------------------- /neuralpredictors/training/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/training/early_stopping.py -------------------------------------------------------------------------------- /neuralpredictors/training/tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/training/tracking.py -------------------------------------------------------------------------------- /neuralpredictors/training/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/training/utils.py -------------------------------------------------------------------------------- /neuralpredictors/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/neuralpredictors/utils.py -------------------------------------------------------------------------------- /notebooks/2dcores_starter.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/notebooks/2dcores_starter.ipynb -------------------------------------------------------------------------------- /notebooks/example_sysident_models.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/notebooks/example_sysident_models.ipynb -------------------------------------------------------------------------------- /notebooks/readouts_tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/notebooks/readouts_tutorial.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- 1 | black==22.1.0 2 | mypy==0.910 3 | isort==5.10.1 4 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/setup.py -------------------------------------------------------------------------------- /test/layers/cores/test_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/test/layers/cores/test_conv2d.py -------------------------------------------------------------------------------- /test/test_TransferLearningCore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/test/test_TransferLearningCore.py -------------------------------------------------------------------------------- /test/test_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/test/test_constraints.py -------------------------------------------------------------------------------- /test/test_training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/test_training/test_early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/test/test_training/test_early_stopping.py -------------------------------------------------------------------------------- /test/utils_for_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sinzlab/neuralpredictors/HEAD/test/utils_for_tests.py --------------------------------------------------------------------------------