From d0418b9efaf6fe959a52f4be54bb897eb6b616d3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ren=C3=A9=20Knaebel?= Date: Wed, 12 Jul 2017 15:18:45 +0200 Subject: [PATCH] fix hyperband adding h5data and remove param sample --- main.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/main.py b/main.py index 86a8bcb..e7f8c98 100644 --- a/main.py +++ b/main.py @@ -103,14 +103,10 @@ def main_hyperband(): "kernels_main": [1, 3, 5, 7, 9], "dense_main": [16, 32, 64, 128, 256, 512], } - param = hyperband.sample_params(params) - logger.info(param) logger.info("create training dataset") - domain_tr, flow_tr, client_tr, server_tr = dataset.create_dataset_from_flows(user_flow_df, char_dict, - max_len=args.domain_length, - window_size=args.window) - + domain_tr, flow_tr, client_tr, server_tr = load_or_generate_h5data(args.h5data, args.train_data, + args.domain_length, args.window) hp = hyperband.Hyperband(params, [domain_tr, flow_tr], [client_tr, server_tr])