Index

A

act() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) (aind_dynamic_foraging_models.generative_model.forager_loss_counting.ForagerLossCounting method) (aind_dynamic_foraging_models.generative_model.forager_q_learning.ForagerQLearning method) act_epsilon_greedy() (in module aind_dynamic_foraging_models.generative_model.act_functions) act_loss_counting() (in module aind_dynamic_foraging_models.generative_model.act_functions) act_softmax() (in module aind_dynamic_foraging_models.generative_model.act_functions) aind_dynamic_foraging_models module aind_dynamic_foraging_models.generative_model module aind_dynamic_foraging_models.generative_model.act_functions module aind_dynamic_foraging_models.generative_model.base module aind_dynamic_foraging_models.generative_model.forager_loss_counting module aind_dynamic_foraging_models.generative_model.forager_q_learning module

aind_dynamic_foraging_models.generative_model.foragers module aind_dynamic_foraging_models.generative_model.learn_functions module aind_dynamic_foraging_models.generative_model.params module aind_dynamic_foraging_models.generative_model.params.forager_loss_counting_params module aind_dynamic_foraging_models.generative_model.params.forager_q_learning_params module aind_dynamic_foraging_models.generative_model.params.util module aind_dynamic_foraging_models.logistic_regression module aind_dynamic_foraging_models.logistic_regression.model module aind_dynamic_foraging_models.logistic_regression.plot module

B

biasL (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute)

C

choice_kernel_relative_weight (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute) choice_kernel_step_size (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute)

choose_ps() (in module aind_dynamic_foraging_models.generative_model.act_functions) create_pydantic_models_dynamic() (in module aind_dynamic_foraging_models.generative_model.params.util)

D

DynamicForagingAgentMLEBase (class in aind_dynamic_foraging_models.generative_model.base)

E

epsilon (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute)

exp_func() (in module aind_dynamic_foraging_models.logistic_regression.model)

F

fit() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) fit_logistic_regression() (in module aind_dynamic_foraging_models.logistic_regression.model) FORAGER_CLASSES (aind_dynamic_foraging_models.generative_model.foragers.ForagerCollection attribute) FORAGER_PRESETS (aind_dynamic_foraging_models.generative_model.foragers.ForagerCollection attribute)

ForagerCollection (class in aind_dynamic_foraging_models.generative_model.foragers) ForagerLossCounting (class in aind_dynamic_foraging_models.generative_model.forager_loss_counting) ForagerQLearning (class in aind_dynamic_foraging_models.generative_model.forager_q_learning) forget_rate_unchosen (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute)

G

generate_pydantic_loss_counting_params() (in module aind_dynamic_foraging_models.generative_model.params.forager_loss_counting_params) generate_pydantic_q_learning_params() (in module aind_dynamic_foraging_models.generative_model.params.forager_q_learning_params) get_agent_alias() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) (aind_dynamic_foraging_models.generative_model.forager_loss_counting.ForagerLossCounting method) (aind_dynamic_foraging_models.generative_model.forager_q_learning.ForagerQLearning method) get_agent_class() (aind_dynamic_foraging_models.generative_model.foragers.ForagerCollection method) get_all_foragers() (aind_dynamic_foraging_models.generative_model.foragers.ForagerCollection method) get_choice_history() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) get_fitting_result_dict() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method)

get_forager() (aind_dynamic_foraging_models.generative_model.foragers.ForagerCollection method) get_latent_variables() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) (aind_dynamic_foraging_models.generative_model.forager_loss_counting.ForagerLossCounting method) (aind_dynamic_foraging_models.generative_model.forager_q_learning.ForagerQLearning method) get_p_reward() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) get_params() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) get_params_options() (in module aind_dynamic_foraging_models.generative_model.params.util) get_params_str() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) get_preset_forager() (aind_dynamic_foraging_models.generative_model.foragers.ForagerCollection method) get_reward_history() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method)

I

is_preset() (aind_dynamic_foraging_models.generative_model.foragers.ForagerCollection method)

L

learn() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) (aind_dynamic_foraging_models.generative_model.forager_loss_counting.ForagerLossCounting method) (aind_dynamic_foraging_models.generative_model.forager_q_learning.ForagerQLearning method) learn_choice_kernel() (in module aind_dynamic_foraging_models.generative_model.learn_functions) learn_loss_counting() (in module aind_dynamic_foraging_models.generative_model.learn_functions)

learn_rate (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute) learn_rate_rew (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute) learn_rate_unrew (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute) learn_RWlike() (in module aind_dynamic_foraging_models.generative_model.learn_functions) loss_count_threshold_mean (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute) loss_count_threshold_std (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute)

M

module aind_dynamic_foraging_models aind_dynamic_foraging_models.generative_model aind_dynamic_foraging_models.generative_model.act_functions aind_dynamic_foraging_models.generative_model.base aind_dynamic_foraging_models.generative_model.forager_loss_counting aind_dynamic_foraging_models.generative_model.forager_q_learning aind_dynamic_foraging_models.generative_model.foragers aind_dynamic_foraging_models.generative_model.learn_functions aind_dynamic_foraging_models.generative_model.params aind_dynamic_foraging_models.generative_model.params.forager_loss_counting_params aind_dynamic_foraging_models.generative_model.params.forager_q_learning_params aind_dynamic_foraging_models.generative_model.params.util aind_dynamic_foraging_models.logistic_regression aind_dynamic_foraging_models.logistic_regression.model aind_dynamic_foraging_models.logistic_regression.plot

N

negLL() (in module aind_dynamic_foraging_models.generative_model.base)

P

ParamsSymbols (class in aind_dynamic_foraging_models.generative_model.params) perform() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) perform_closed_loop() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) perform_closed_loop_multi_session() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) plot_fitted_session() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method)

plot_latent_variables() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) (aind_dynamic_foraging_models.generative_model.forager_loss_counting.ForagerLossCounting method) (aind_dynamic_foraging_models.generative_model.forager_q_learning.ForagerQLearning method) plot_logistic_regression() (in module aind_dynamic_foraging_models.logistic_regression.plot) plot_session() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method) prepare_logistic_design_matrix() (in module aind_dynamic_foraging_models.logistic_regression.model)

R

reset_to_threshold (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute)

S

set_params() (aind_dynamic_foraging_models.generative_model.base.DynamicForagingAgentMLEBase method)

softmax() (in module aind_dynamic_foraging_models.generative_model.act_functions) softmax_inverse_temperature (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute)

T

threshold (aind_dynamic_foraging_models.generative_model.params.ParamsSymbols attribute)