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
update biphasics model #617
base: master
Are you sure you want to change the base?
update biphasics model #617
Conversation
|
||
# bright | ||
F_bright = a2 * scaled_amps + a3 * freq | ||
if self.amp_cutoff: |
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.
dont need amp cutoff
setattr(self, key, val) | ||
self.is_built = True | ||
|
||
def __getattr__(self, attr): |
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.
A lot of these are not needed her, since theyre in BiphasicAxonMapModel
# for simulation based on the current set of parameters. | ||
print("Model built with current parameters.") | ||
|
||
def _predict_spatial(self, earray, stim): |
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.
Any logic needed here can be moved to biphasicaxonmap
Description
Please include a summary of the change and which issue it closes below. Include relevant motivation and context.
This is a test PR to check the implementation of pytorch version of granley2021 biphasics map model. Originally it is a jax model. The existing code is subject to change after running the tests.
Closes #(issue)
Type of Change
Please delete options that are not relevant.
Checklist
For detailed information on these and other aspects, see the contribution guidelines: https://pulse2percept.readthedocs.io/en/latest/developers/contributing.html