8201900250bab24c768542f80090dc6381f05af1,niftynet/layer/loss_segmentation.py,LossFunction,__init__,#LossFunction#Any#Any#Any#Any#,21

Before Change

        super(LossFunction, self).__init__(name=name)
        self._num_classes = n_class
        if loss_func_params is not None:
            self._loss_func_params = loss_func_params
        else:
            self._loss_func_params = {}
        self._data_loss_func = None
        self.make_callable_loss_func(loss_type)

    def make_callable_loss_func(self, type_str):
        self._data_loss_func = LossSegmentationFactory.create(type_str)

After Change

        // set loss function and function-specific additional params.
        self._data_loss_func = LossSegmentationFactory.create(loss_type)
        self._loss_func_params = \
            loss_func_params if loss_func_params is not None else dict()

    def layer_op(self,
                 prediction,
                 ground_truth,

Instances