Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add reset! function for Agent and AbstractEnvironment #340

Merged
merged 5 commits into from
Feb 15, 2024
Merged

Conversation

harisorgn
Copy link
Member

Adds a function to reset an Agent's parameters, mainly the weights that are learned but also auxiliary parameters (see ./test/reinforcement_learning.jl for all parameters that change during RL).

The dispatch for AbstractEnvironment resets the current_trial to 1, which affects the stimulus that is active.

cc @MasonProtter

@harisorgn harisorgn merged commit fe808d8 into master Feb 15, 2024
2 checks passed
@harisorgn harisorgn deleted the ho/reset branch February 15, 2024 14:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant