├── .flake8 ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE.md ├── README.md ├── docs ├── failed_training.png ├── model_schematic.png └── successful_training.png ├── environment.yml ├── manuscript ├── __init__.py ├── figure2.py ├── figure3.py ├── figure4.py ├── figure5.py ├── figureS10.py ├── figureS2.py ├── figureS3.py ├── figureS4.py ├── figureS5.py ├── figureS6.py ├── figureS7.py ├── figureS8.py ├── figureS9.py ├── make_main.py ├── make_supplement.py └── preprocessing.py ├── pyproject.toml ├── task_dyva ├── __init__.py ├── config │ ├── __init__.py │ └── model_config.yaml ├── experiment.py ├── logging.py ├── model_analysis.py ├── models │ ├── __init__.py │ └── lldvae.py ├── objectives.py ├── taskdataset.py ├── testing.py ├── transforms.py ├── utils.py └── visualization.py ├── tests ├── __init__.py ├── test_data │ ├── user1089 │ │ └── user1089.pickle │ ├── user1365 │ │ └── user1365.pickle │ └── user2139 │ │ └── user2139.pickle ├── test_ebb_flow_dataset.py ├── test_ebb_flow_game_data.py ├── test_experiment.py ├── test_model.py └── test_resampling.py └── train.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/README.md -------------------------------------------------------------------------------- /docs/failed_training.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/docs/failed_training.png -------------------------------------------------------------------------------- /docs/model_schematic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/docs/model_schematic.png -------------------------------------------------------------------------------- /docs/successful_training.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/docs/successful_training.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/environment.yml -------------------------------------------------------------------------------- /manuscript/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /manuscript/figure2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figure2.py -------------------------------------------------------------------------------- /manuscript/figure3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figure3.py -------------------------------------------------------------------------------- /manuscript/figure4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figure4.py -------------------------------------------------------------------------------- /manuscript/figure5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figure5.py -------------------------------------------------------------------------------- /manuscript/figureS10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS10.py -------------------------------------------------------------------------------- /manuscript/figureS2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS2.py -------------------------------------------------------------------------------- /manuscript/figureS3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS3.py -------------------------------------------------------------------------------- /manuscript/figureS4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS4.py -------------------------------------------------------------------------------- /manuscript/figureS5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS5.py -------------------------------------------------------------------------------- /manuscript/figureS6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS6.py -------------------------------------------------------------------------------- /manuscript/figureS7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS7.py -------------------------------------------------------------------------------- /manuscript/figureS8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS8.py -------------------------------------------------------------------------------- /manuscript/figureS9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/figureS9.py -------------------------------------------------------------------------------- /manuscript/make_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/make_main.py -------------------------------------------------------------------------------- /manuscript/make_supplement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/make_supplement.py -------------------------------------------------------------------------------- /manuscript/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/manuscript/preprocessing.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/pyproject.toml -------------------------------------------------------------------------------- /task_dyva/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/__init__.py -------------------------------------------------------------------------------- /task_dyva/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /task_dyva/config/model_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/config/model_config.yaml -------------------------------------------------------------------------------- /task_dyva/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/experiment.py -------------------------------------------------------------------------------- /task_dyva/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/logging.py -------------------------------------------------------------------------------- /task_dyva/model_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/model_analysis.py -------------------------------------------------------------------------------- /task_dyva/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .lldvae import LLDVAE 2 | -------------------------------------------------------------------------------- /task_dyva/models/lldvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/models/lldvae.py -------------------------------------------------------------------------------- /task_dyva/objectives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/objectives.py -------------------------------------------------------------------------------- /task_dyva/taskdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/taskdataset.py -------------------------------------------------------------------------------- /task_dyva/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/testing.py -------------------------------------------------------------------------------- /task_dyva/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/transforms.py -------------------------------------------------------------------------------- /task_dyva/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/utils.py -------------------------------------------------------------------------------- /task_dyva/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/task_dyva/visualization.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_data/user1089/user1089.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/tests/test_data/user1089/user1089.pickle -------------------------------------------------------------------------------- /tests/test_data/user1365/user1365.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/tests/test_data/user1365/user1365.pickle -------------------------------------------------------------------------------- /tests/test_data/user2139/user2139.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/tests/test_data/user2139/user2139.pickle -------------------------------------------------------------------------------- /tests/test_ebb_flow_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/tests/test_ebb_flow_dataset.py -------------------------------------------------------------------------------- /tests/test_ebb_flow_game_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/tests/test_ebb_flow_game_data.py -------------------------------------------------------------------------------- /tests/test_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/tests/test_experiment.py -------------------------------------------------------------------------------- /tests/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/tests/test_model.py -------------------------------------------------------------------------------- /tests/test_resampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/tests/test_resampling.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pauljaffe/task-dyva/HEAD/train.py --------------------------------------------------------------------------------