Exemplo n.º 1
0
 def __init__(self, imputer, loss='cross entropy'):
     self.imputer = imputer
     self.loss_fn = utils.get_loss(loss, reduction='none')
Exemplo n.º 2
0
 def __init__(self, model, imputer, loss):
     self.model = model
     self.imputer = imputer
     self.loss_fn = utils.get_loss(loss, reduction='none')