volver a poner 256 Fully-Connected

lytex · Sep 5, 2024 · 393d987 · 393d987
1 parent 0f979fd
commit 393d987
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/02_entrenamiento.py b/02_entrenamiento.py
@@ -252,13 +252,13 @@ def gen_model_2_levels(inputs, classes, activation = 'relu',summary=False, binar
     model_f = concatenate([m.output for m in net["global_par"]] + [m.output for m in net["global_impar"]] + [m.output for m in net["local_par"]] + [m.output for m in net["local_impar"]], axis=-1)
     model_f = BatchNormalization(axis=-1)(model_f)
     model_f = Dropout(dropout)(model_f)
-    model_f = Dense(512,activation=activation, kernel_regularizer=L1L2( l1=l1, l2=l2))(model_f)
+    model_f = Dense(256,activation=activation, kernel_regularizer=L1L2( l1=l1, l2=l2))(model_f)
     model_f = Dropout(dropout)(model_f)
-    model_f = Dense(512,activation=activation, kernel_regularizer=L1L2( l1=l1, l2=l2,))(model_f)
+    model_f = Dense(256,activation=activation, kernel_regularizer=L1L2( l1=l1, l2=l2,))(model_f)
     model_f = Dropout(dropout)(model_f)
-    model_f = Dense(512,activation=activation, kernel_regularizer=L1L2( l1=l1, l2=l2,))(model_f)
+    model_f = Dense(256,activation=activation, kernel_regularizer=L1L2( l1=l1, l2=l2,))(model_f)
     model_f = Dropout(dropout)(model_f)
-    model_f = Dense(512,activation=activation, kernel_regularizer=L1L2( l1=l1, l2=l2,))(model_f)
+    model_f = Dense(256,activation=activation, kernel_regularizer=L1L2( l1=l1, l2=l2,))(model_f)
     if binary_classification:
         model_f = Dense(1,activation='sigmoid', name="sigmoid")(model_f)
     else: