From fa38a43d30da9481c15b76ea35f957aec368ab99 Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Mon, 30 Apr 2018 08:42:33 +0200
Subject: [PATCH] dump options to json dict
---
toolkit.py | 34 ++++++++++++++++++++--------------
1 file changed, 20 insertions(+), 14 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 59b08b7..eb5b02c 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -47,20 +47,26 @@ class KerasROOTClassification(object):
dataset_names = ["x_train", "x_test", "y_train", "y_test", "w_train", "w_test"]
- def __init__(self, name,
- signal_trees, bkg_trees, branches, weight_expr, identifiers,
- selection=None,
- layers=3,
- nodes=64,
- batch_size=128,
- validation_split=0.33,
- activation_function='relu',
- out_dir="./outputs",
- scaler_type="RobustScaler",
- step_signal=2,
- step_bkg=2,
- optimizer="SGD",
- optimizer_opts=None):
+ def __init__(self, name, *args, **kwargs):
+ self._init_from_args(name, *args, **kwargs)
+ with open(os.path.join(self.project_dir, "options.json"), "w") as of:
+ json.dump(dict(args=args, kwargs=kwargs), of)
+
+
+ def _init_from_args(self, name,
+ signal_trees, bkg_trees, branches, weight_expr, identifiers,
+ selection=None,
+ layers=3,
+ nodes=64,
+ batch_size=128,
+ validation_split=0.33,
+ activation_function='relu',
+ out_dir="./outputs",
+ scaler_type="RobustScaler",
+ step_signal=2,
+ step_bkg=2,
+ optimizer="SGD",
+ optimizer_opts=None):
self.name = name
self.signal_trees = signal_trees
self.bkg_trees = bkg_trees
--
GitLab