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

wip

Browse Source
This commit is contained in:
René Knaebel 2017-07-17 08:44:58 +02:00
parent d33c9f44ec
commit 79fc441fe1
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dataset.py
View File

@ -157,9 +157,6 @@ def create_dataset_from_lists(chunks, vocab, max_len):
logger.info(" compute domain features") logger.info(" compute domain features")
domain_features = [] domain_features = []
for ds in tqdm(map(lambda f: f.domain, chunks)): for ds in tqdm(map(lambda f: f.domain, chunks)):
# TODO: fix this correctly
# assert min(np.atleast_3d(ds).shape) > 0, f"shape of 0 for {ds}"
if not ds: continue
domain_features.append(np.apply_along_axis(get_domain_features_reduced, 2, np.atleast_3d(ds))) domain_features.append(np.apply_along_axis(get_domain_features_reduced, 2, np.atleast_3d(ds)))
domain_features = np.concatenate(domain_features, 0) domain_features = np.concatenate(domain_features, 0)
logger.info(" compute flow features") logger.info(" compute flow features")