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

extract weighting function

Browse Source
This commit is contained in:
René Knaebel
2017-11-10 10:18:13 +01:00
parent 9ce11e4db4
commit 4fc2f0c925
3 changed files with 45 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
models/__init__.py
View File

@@ -1,8 +1,17 @@
from collections import namedtuple
from keras.models import Model
from . import networks
from .metrics import *
NetworkParameters = namedtuple("NetworkParameters", [
"type", "flow_features", "window_size", "domain_length", "output",
"embedding_size",
"domain_filter", "domain_kernel", "domain_dense", "domain_dropout",
"main_filter", "main_kernel", "main_dense", "main_dropout",
])
def create_model(model, output_type):
if output_type == "both":
@@ -14,6 +23,7 @@ def create_model(model, output_type):
def get_models_by_params(params: dict):
K.clear_session()
# decomposing param section
# mainly embedding model
network_type = params.get("type")