.coveragerc-hb
.flake8
.gitignore
.readthedocs.yaml
LICENSE
README.md
classes_aind_dynamic_foraging_models.png
classes_aind_dynamic_foraging_models.svg
packages_aind_dynamic_foraging_models.png
pyproject.toml
setup.py
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/ISSUE_TEMPLATE/user-story.md
.github/workflows/init.yml
.github/workflows/tag_and_publish.yml
.github/workflows/test_and_lint.yml
docs/Makefile
docs/make.bat
docs/source/RL_models.rst
docs/source/aind_dynamic_foraging_models.generative_model.params.rst
docs/source/aind_dynamic_foraging_models.generative_model.rst
docs/source/aind_dynamic_foraging_models.hierarchical_bayes.rst
docs/source/aind_dynamic_foraging_models.logistic_regression.rst
docs/source/aind_dynamic_foraging_models.rst
docs/source/conf.py
docs/source/index.rst
docs/source/logistic_models.rst
docs/source/modules.rst
docs/source/_static/dark-logo.svg
docs/source/_static/favicon.ico
docs/source/_static/light-logo.svg
notebook/demo_RL_agents-compare_threshold.ipynb
notebook/demo_RL_agents-compare_threshold_noreset_asymmetricLR.ipynb
notebook/demo_RL_agents.ipynb
notebook/demo_gym_env.ipynb
notebook/demo_hierarchical_bayes.ipynb
notebook/demo_logistic_regression.ipynb
notebook/demo_rl_mle_fitting_new_test_time.ipynb
notebook/model_fitting-compare_to_threshold.ipynb
src/aind_dynamic_foraging_models/__init__.py
src/aind_dynamic_foraging_models.egg-info/PKG-INFO
src/aind_dynamic_foraging_models.egg-info/SOURCES.txt
src/aind_dynamic_foraging_models.egg-info/dependency_links.txt
src/aind_dynamic_foraging_models.egg-info/requires.txt
src/aind_dynamic_foraging_models.egg-info/top_level.txt
src/aind_dynamic_foraging_models/generative_model/__init__.py
src/aind_dynamic_foraging_models/generative_model/act_functions.py
src/aind_dynamic_foraging_models/generative_model/base.py
src/aind_dynamic_foraging_models/generative_model/forager_actor_critic.py
src/aind_dynamic_foraging_models/generative_model/forager_compare_threshold.py
src/aind_dynamic_foraging_models/generative_model/forager_loss_counting.py
src/aind_dynamic_foraging_models/generative_model/forager_q_learning.py
src/aind_dynamic_foraging_models/generative_model/foragers.py
src/aind_dynamic_foraging_models/generative_model/learn_functions.py
src/aind_dynamic_foraging_models/generative_model/params/__init__.py
src/aind_dynamic_foraging_models/generative_model/params/forager_actor_critic_params.py
src/aind_dynamic_foraging_models/generative_model/params/forager_compare_threshold_params.py
src/aind_dynamic_foraging_models/generative_model/params/forager_loss_counting_params.py
src/aind_dynamic_foraging_models/generative_model/params/forager_q_learning_params.py
src/aind_dynamic_foraging_models/generative_model/params/util.py
src/aind_dynamic_foraging_models/hierarchical_bayes/__init__.py
src/aind_dynamic_foraging_models/hierarchical_bayes/artifacts.py
src/aind_dynamic_foraging_models/hierarchical_bayes/heldout.py
src/aind_dynamic_foraging_models/hierarchical_bayes/likelihood.py
src/aind_dynamic_foraging_models/hierarchical_bayes/model.py
src/aind_dynamic_foraging_models/hierarchical_bayes/plotting.py
src/aind_dynamic_foraging_models/hierarchical_bayes/two_stage.py
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/RESULTS.md
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/benchmark_stan_vs_numpyro.py
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/geometry_experiment.py
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/lane_scaling.py
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/reference_stan/README.md
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/reference_stan/stan_qLearning_5params.stan
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/slurm/README.md
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/slurm/cpu16.sbatch
src/aind_dynamic_foraging_models/hierarchical_bayes/benchmarks/slurm/gpu.sbatch
src/aind_dynamic_foraging_models/hierarchical_bayes/docs/design-decisions.md
src/aind_dynamic_foraging_models/hierarchical_bayes/recovery/estimator_comparison.py
src/aind_dynamic_foraging_models/logistic_regression/__init__.py
src/aind_dynamic_foraging_models/logistic_regression/model.py
src/aind_dynamic_foraging_models/logistic_regression/plot.py
src/aind_dynamic_foraging_models/logistic_regression/readme.md
tests/__init__.py
tests/_hb_deps.py
tests/test_ActorCritic.py
tests/test_Bari.py
tests/test_CompareToThreshold.py
tests/test_Hattori.py
tests/test_RescorlaWagner.py
tests/test_get_forager.py
tests/test_hb_artifacts.py
tests/test_hb_hattori_model.py
tests/test_hb_hattori_parity.py
tests/test_hb_heldout.py
tests/test_hb_heldout_batched.py
tests/test_hb_packaging.py
tests/test_hb_plotting.py
tests/test_hb_three_level.py
tests/test_hb_two_stage.py
tests/test_hb_value_trajectory.py
tests/test_logistic_regression.py
tests/test_loss_counting.py
tests/test_multi_session_fitting.py
tests/test_pydantic_models.py