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

fix flat 2 model bug caused by changing input layer

This commit is contained in:
René Knaebel 2017-10-06 16:36:58 +02:00
parent 5741f8ee0e
commit 0b26c6125c
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
arguments.py
View File

@ -35,7 +35,7 @@ parser.add_argument("--type", action="store", dest="model_type",
default="final") # inter, final, staggered default="final") # inter, final, staggered
parser.add_argument("--depth", action="store", dest="model_depth", parser.add_argument("--depth", action="store", dest="model_depth",
default="small") # small, medium default="flat1") # small, medium
parser.add_argument("--model_output", action="store", dest="model_output", parser.add_argument("--model_output", action="store", dest="model_output",
default="both") default="both")

6
models/flat_2.py
View File

@ -41,9 +41,9 @@ def get_model(cnnDropout, flow_features, domain_features, window_size, domain_le
ipt_domains = Input(shape=(window_size, domain_length), name="ipt_domains") ipt_domains = Input(shape=(window_size, domain_length), name="ipt_domains")
encoded = TimeDistributed(cnn, name="domain_cnn")(ipt_domains) encoded = TimeDistributed(cnn, name="domain_cnn")(ipt_domains)
ipt_flows = Input(shape=(window_size, flow_features), name="ipt_flows") ipt_flows = Input(shape=(window_size, flow_features), name="ipt_flows")
ipt_flows = BatchNormalization()(ipt_flows) y = BatchNormalization()(ipt_flows)
ipt_flows = Dense(dense_dim, activation=selu)(ipt_flows) y = Dense(dense_dim, activation=selu)(y)
merged = keras.layers.concatenate([encoded, ipt_flows], -1) merged = keras.layers.concatenate([encoded, y], -1)
# CNN processing a small slides of flow windows # CNN processing a small slides of flow windows
y = Conv1D(cnn_dims, y = Conv1D(cnn_dims,
kernel_size, kernel_size,