Modified Dense Layer @ Lines 36 & 71

Palashio · Jul 18, 2020 · fe1704a · fe1704a
1 parent 86bf214
commit fe1704a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libra/modeling/prediction_model_creation.py b/libra/modeling/prediction_model_creation.py
@@ -33,7 +33,7 @@ def get_keras_model_reg(dataset, i):
         else:
             model.add(
                 Dense(
-                    i * 64,
+                    64,
                     kernel_initializer="normal",
                     activation="relu"))
 
@@ -68,7 +68,7 @@ def get_keras_model_class(dataset, i, num_classes):
         else:
             model.add(
                 Dense(
-                    i * 64,
+                    64,
                     kernel_initializer="normal",
                     activation="relu"))