-
Notifications
You must be signed in to change notification settings - Fork 3
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
feat: use external training state #18
Conversation
Unit Test Results34 tests - 1 34 ✔️ - 1 1m 3s ⏱️ +16s Results for commit be24cda. ± Comparison against base commit eeb9bcb. This pull request removes 9 and adds 8 tests. Note that renamed tests count towards both.
♻️ This comment has been updated with latest results. |
* test: updated convergence catx and contest * fix: fix mypy type error * test: updated test_tree * test: updated network module * docs: updated getting started guide * chore: answered review comments
tests/test_catx.py
Outdated
if not request: | ||
action_min = 0.0 | ||
action_max = 1.0 | ||
else: | ||
action_min = request.param[0] | ||
action_max = request.param[1] | ||
|
||
return mk # type: ignore | ||
catx = CATX( | ||
catx_network=catx_network_with_dropout_extras, | ||
optimizer=optax.adam(learning_rate=0.01), | ||
discretization_parameter=4, | ||
bandwidth=1.5 / 4, | ||
action_min=action_min, | ||
action_max=action_max, | ||
) | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
we can probably create a function that takes in the catx_network as parameter as well as the other fixture parameters and returns the CATX object instanciated. Since the fixtures catx and catx_with_dropout_extras differ only by the class used for the catx network.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM - nice work!
No description provided.