rene/ma_cisco_malware
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

remove reguliarizer on dense too

Browse Source
This commit is contained in:
René Knaebel 2017-09-11 16:48:49 +02:00
parent 3a44efa775
commit 6ce8fb464f
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/pauls_networks.py
View File

@ -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) encoded = TimeDistributed(cnn, name="domain_cnn")(ipt_domains)
merged = keras.layers.concatenate([encoded, ipt_flows], -1) merged = keras.layers.concatenate([encoded, ipt_flows], -1)
y = Dense(dense_dim, y = Dense(dense_dim,
kernel_regularizer=l2(0.1),
activation="relu", activation="relu",
name="dense_server")(merged) name="dense_server")(merged)
out_server = Dense(1, activation="sigmoid", name="server")(y) 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 = GlobalMaxPooling1D()(y)
y = Dropout(dropout)(y) y = Dropout(dropout)(y)
y = Dense(dense_dim, y = Dense(dense_dim,
kernel_regularizer=l2(0.1),
activation='relu', activation='relu',
name="dense_client")(y) name="dense_client")(y)