3ec1dac0608e511d4cf28e93a3fb908bbabceac5,nn/loss.py,LossFunction,set_loss_type,#LossFunction#Any#,15

Before Change

            self.data_loss_fun = dice
        elif type_str == "GDSC":
            self.data_loss_fun = GDSC_loss
        elif type_str == "SensSpec":
            self.data_loss_fun = sensitivity_specificity_loss

    def set_regularisation_type(self, type_str):

After Change

            loss_type, decay, reg_type))

    def set_loss_type(self, type_str):
        accepted_functions = {"CrossEntropy": cross_entropy,
                              "Dice": dice,
                              "GDSC": GDSC_loss,
                              "SensSpec": sensitivity_specificity_loss}
        if type_str in accepted_functions.keys():
            self.data_loss_fun = accepted_functions[type_str]
        else:
            edit_distances = {}
            for loss_name in accepted_functions.keys():

Instances