From d6df0f2b9b5e052f3b454e43bd6791940217f19e Mon Sep 17 00:00:00 2001 From: Ben Bolte Date: Mon, 15 Aug 2016 13:09:00 -0700 Subject: [PATCH] fixed minor bug --- insurance_qa_eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/insurance_qa_eval.py b/insurance_qa_eval.py index b4656a1..5a0cb21 100644 --- a/insurance_qa_eval.py +++ b/insurance_qa_eval.py @@ -16,7 +16,7 @@ random.seed(42) class Evaluator: - def __init__(self, conf, model=None, optimizer=None): + def __init__(self, conf, model, optimizer=None): try: data_path = os.environ['INSURANCE_QA'] except KeyError: @@ -24,7 +24,7 @@ class Evaluator: sys.exit(1) if isinstance(conf, str): conf = json.load(open(conf, 'rb')) - self.model = conf['model'](conf) if model is None else model + self.model = model(conf) self.path = data_path self.conf = conf self.params = conf['training']