Logistic regression models¶
See this demo notebook.
Choosing logistic regression models¶
Su 2022¶
Bari 2019¶
Hattori 2019¶
Miller 2021¶
Encodings¶
Ignored trials are removed
choice |
reward |
Choice |
Reward |
RewardedChoice |
UnrewardedChoice |
Choice * Reward |
|---|---|---|---|---|---|---|
L |
yes |
-1 |
1 |
-1 |
0 |
-1 |
L |
no |
-1 |
-1 |
0 |
-1 |
1 |
R |
yes |
1 |
1 |
1 |
0 |
1 |
L |
yes |
-1 |
1 |
-1 |
0 |
-1 |
R |
no |
1 |
-1 |
0 |
1 |
-1 |
R |
yes |
1 |
1 |
1 |
0 |
1 |
L |
no |
-1 |
-1 |
0 |
-1 |
1 |
Some observations:
RewardedChoiceandUnrewardedChoiceare orthogonal.Choice = RewardedChoice + UnrewardedChoiceChoice * Reward = RewardedChoice - UnrewardedChoice
Comparison¶
Su 2022 |
Bari 2019 |
Hattori 2019 |
Miller 2021 |
|
|---|---|---|---|---|
Equivalent to |
RewC + UnrC |
RewC + (RewC + UnrC) |
RewC + UnrC + (RewC + UnrC) |
(RewC + UnrC) + (RewC - UnrC) + Rew |
Severity of multicollinea- rity |
Not at all |
Medium |
Severe |
Slight |
Interpretation |
Like a RL model with different learning rates on reward and unreward- ed trials |
Like a RL model that only updates on rewarded trials, plus a choice kernel |
Like a RL model that has different learning rates on reward and unrewarded trials, plus a choice kernel |
Like a RL model that has symmetric learning rates for rewarded and unrewarded trials, plus a choice |
Regularization and optimization¶
The choice of optimizer depends on the penalty term, as listed here.
lbfgs- [l2, None]liblinear- [l1,l2]newton-cg- [l2, None]newton-cholesky- [l2, None]sag- [l2, None]saga- [elasticnet,l1,l2, None]
See also¶
Foraging model simulation, model recovery, etc.: https://github.com/hanhou/Dynamic-Foraging