def __init__(self, imputer, loss='cross entropy'):
self.imputer = imputer
self.loss_fn = utils.get_loss(loss, reduction='none')
def __init__(self, model, imputer, loss):
self.model = model
self.imputer = imputer
self.loss_fn = utils.get_loss(loss, reduction='none')