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

indel support: predict_mutations_on_batch() #6

Open
krrome opened this issue Jun 13, 2018 · 0 comments
Open

indel support: predict_mutations_on_batch() #6

krrome opened this issue Jun 13, 2018 · 0 comments

Comments

@krrome
Copy link
Contributor

krrome commented Jun 13, 2018

In analogy to predict_on_batch() kipoi_veff should enable a predict_mutations_on_batch() function that will use the inputs_ref, inputs_alt (inputs_ref_rc, inputs_alt_rc) keys returned from the MutationDatasetMixinobject to produce 2 (or 4) model prediction outputs.

This may be designed as a mixin for kipoi.model.Model classes:

class PredictMutationsMixin(object):
    def predict_mutations_on_batch(self, x):
        return [self.predict_on_batch(x[k]) for k in x if k.startswith("inputs_")]
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

No branches or pull requests

1 participant