Fixed bug.

ai2es · Feb 28, 2024 · 500f509 · 500f509
1 parent e2ccc0e
commit 500f509
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlguess/keras/models_deprecated.py b/mlguess/keras/models_deprecated.py
@@ -9,7 +9,7 @@
 from tensorflow.keras.layers import Dense, LeakyReLU, GaussianNoise, Dropout
 from tensorflow.keras.optimizers import Adam, SGD
 from mlguess.keras.layers import DenseNormalGamma, DenseNormal
-from mlguess.keras.losses import EvidentialRegressionLoss, EvidentialRegressionCoupledLoss, GaussianNLL
+from mlguess.keras.losses import EvidentialRegressionLoss, EvidentialRegressionCoupledLoss, gaussian_nll
 from mlguess.keras.losses import DirichletEvidentialLoss
 from mlguess.keras.callbacks import ReportEpoch
 from imblearn.under_sampling import RandomUnderSampler
@@ -597,7 +597,7 @@ def build_neural_network(self, inputs, outputs):
             inputs (int): Number of input predictor variables
             outputs (int): Number of output predictor variables
         """
-        self.loss = GaussianNLL
+        self.loss = gaussian_nll
 
         nn_input = Input(shape=(inputs,), name="input")
         nn_model = nn_input