Enable/Disable inferring output neuron function

LukasZahradnik · Jul 2, 2022 · e95302d · e95302d
1 parent fbf8df0
commit e95302d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/neuralogic/core/settings/settings_proxy.py b/neuralogic/core/settings/settings_proxy.py
@@ -105,6 +105,8 @@ def error_function(self):
 
     @error_function.setter
     def error_function(self, error_function: ErrorFunction):
+        self.settings.inferOutputNeuronFcn = True
+
         if isinstance(error_function, MSE):
             self.settings.squishLastLayer = False
             self.settings.trainOnlineResultsType = self.settings_class.ResultsType.REGRESSION
@@ -120,6 +122,7 @@ def error_function(self, error_function: ErrorFunction):
                 self.settings.squishLastLayer = True
                 java_error_function = self.settings_class.ErrorFcn.SOFTENTROPY
             else:
+                self.settings.inferOutputNeuronFcn = False
                 self.settings.squishLastLayer = False
                 java_error_function = self.settings_class.ErrorFcn.CROSSENTROPY
         else: