diff --git a/models/pauls_networks.py b/models/pauls_networks.py index 08b4747..0d26ec3 100644 --- a/models/pauls_networks.py +++ b/models/pauls_networks.py @@ -72,7 +72,6 @@ def get_new_model(dropout, flow_features, domain_features, window_size, domain_l encoded = TimeDistributed(cnn, name="domain_cnn")(ipt_domains) merged = keras.layers.concatenate([encoded, ipt_flows], -1) y = Dense(dense_dim, - kernel_regularizer=l2(0.1), activation="relu", name="dense_server")(merged) out_server = Dense(1, activation="sigmoid", name="server")(y) @@ -86,7 +85,6 @@ def get_new_model(dropout, flow_features, domain_features, window_size, domain_l y = GlobalMaxPooling1D()(y) y = Dropout(dropout)(y) y = Dense(dense_dim, - kernel_regularizer=l2(0.1), activation='relu', name="dense_client")(y)