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_")]
In analogy to
predict_on_batch()kipoi_veff should enable apredict_mutations_on_batch()function that will use theinputs_ref,inputs_alt(inputs_ref_rc,inputs_alt_rc) keys returned from theMutationDatasetMixinobject to produce 2 (or 4) model prediction outputs.This may be designed as a mixin for
kipoi.model.Modelclasses: