Пример #1
0
 def __init__(self, imputer, loss='cross entropy'):
     self.imputer = imputer
     self.loss_fn = utils.get_loss(loss, reduction='none')
Пример #2
0
 def __init__(self, model, imputer, loss):
     self.model = model
     self.imputer = imputer
     self.loss_fn = utils.get_loss(loss, reduction='none')