Merge pull request #244 from Abhilash2000/patch-1

Modified Dense Layer @ Lines 36 & 71
Palashio · Jul 18, 2020 · 64392a4 · 64392a4
2 parents 86bf214 + fe1704a
commit 64392a4
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"))