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

WPI

Browse Source
This commit is contained in:
René Knaebel 2017-07-05 19:16:03 +02:00
parent a70d1cb03a
commit 772b07847f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dataset.py
View File

@ -108,12 +108,12 @@ def create_dataset_from_flows(user_flow_df, char_dict, max_len, window_size=10,
for i, user_flow in enumerate(get_flow_per_user(user_flow_df)): for i, user_flow in enumerate(get_flow_per_user(user_flow_df)):
(domain_windows, feature_windows) = get_user_chunks(user_flow, (domain_windows, feature_windows) = get_user_chunks(user_flow,
windowSize=window_size, windowSize=window_size,
overlapping=True, overlapping=False,
maxLengthInSeconds=-1) maxLengthInSeconds=-1)
domains += domain_windows domains += domain_windows
features += feature_windows features += feature_windows
# TODO: remove later # TODO: remove later
if i >= 10: if i >= 50:
break break
print("create training dataset") print("create training dataset")