Skip to content

Commit 7a15f70

Browse files
authored
Merge pull request #118 from AllenNeuralDynamics/han_add_comparethreshold
feat: add comparethreshold
2 parents 0b64bc6 + 34b1027 commit 7a15f70

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

code/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__ver__ = 'v3.2.0'
1+
__ver__ = 'v3.2.1'

code/pages/4_RL model playground.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,8 @@
3030

3131
model_families = {
3232
"Q-learning": "ForagerQLearning",
33-
"Loss counting": "ForagerLossCounting"
33+
"Compare-to-threshold": "ForagerCompareThreshold",
34+
"Loss counting": "ForagerLossCounting",
3435
}
3536

3637
task_families = {
@@ -57,6 +58,7 @@ def select_params(forager):
5758
default_range=[-20, 20],
5859
para_range_override={
5960
"choice_kernel_relative_weight": [0.0, 2.0],
61+
"threshold": [-1.0, 1.0],
6062
})
6163

6264
params = {}

0 commit comments

Comments
 (0)