116e474be75cdcf4c1d53b1c6540097766671b18,ludwig/features/numerical_feature.py,NumericalOutputFeature,__init__,#NumericalOutputFeature#Any#,161

Before Change

class NumericalOutputFeature(NumericalBaseFeature, OutputFeature):
    def __init__(self, feature):
        super().__init__(feature)

        self.loss = {"type": MEAN_SQUARED_ERROR}
        self.clip = None
        self.initializer = None

After Change

class NumericalOutputFeature(NumericalBaseFeature, OutputFeature):
    def __init__(self, feature):
        NumericalBaseFeature.__init__(self, feature)
        OutputFeature.__init__(self, feature)

        self.decoder = "regressor"
        self.loss = {"type": MEAN_SQUARED_ERROR}
        self.clip = None
        self.initializer = None
        self.regularize = True

        decoder_parameters = self.overwrite_defaults(feature)

        self.decoder_obj = self.initialize_decoder(decoder_parameters)

        self._setup_loss()
        self._setup_metrics()

    def logits(
            self,
            inputs,  // hidden

Instances