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

update biphasics model #617

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

HououinKyouma-2036
Copy link

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.

  • New feature (non-breaking change which adds functionality)

Checklist

  • I have performed a self-review of my own code
  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • I have added tests that prove my fix is effective or that my feature works
  • New and existing unit tests pass locally with my changes

For detailed information on these and other aspects, see the contribution guidelines: https://pulse2percept.readthedocs.io/en/latest/developers/contributing.html


# bright
F_bright = a2 * scaled_amps + a3 * freq
if self.amp_cutoff:
Copy link
Member

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):
Copy link
Member

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):
Copy link
Member

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

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.

None yet

2 participants