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

Before Change

        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):

After Change

                 name="loss_function"):

        super(LossFunction, self).__init__(name=name)
        assert n_class > 0, \
            "Number of classes for segmentation loss should be positive."
        self._num_classes = n_class

        // set loss function and function-specific additional params.
        self._data_loss_func = LossSegmentationFactory.create(loss_type)

Instances