示例#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')