def __init__(self, loss=None):
if isinstance(loss, str):
self._loss_type = loss
self._loss_fn = None
self._loss_delta_fn = None
if loss == 'mse':
self._loss_fn = self._loss_mse
self._loss_delta_fn = self._loss_delta_mse
elif loss == 'nll':
self._loss_fn = self._loss_nll
self._loss_delta_fn = self._loss_delta_nll
elif loss:
raise ValueError("unrecognized loss '%s' requested" %
loss_type)
else:
self._loss_type = 'callback'
self._loss_fn = loss[0]
self._loss_delta_fn = loss[1]
self._tmp_E = TempMatrix() # memory for computing loss
self._tmp_e = TempMatrix() # memory for computing loss
def __init__(self):
self._tmp_denom = TempMatrix()
