leaky to leaky_relu

ai2es · Aug 2, 2024 · 1fc7583 · 1fc7583
1 parent 11512f9
commit 1fc7583
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ptype/models.py b/ptype/models.py
@@ -120,7 +120,7 @@ def build_neural_network(self, inputs, outputs):
             inputs (int): Number of input predictor variables
             outputs (int): Number of output predictor variables
         """
-        if self.activation == "leaky":
+        if self.activation == "leaky_relu":
             self.activation = LeakyReLU()
 
         if self.kernel_reg == "l1":