From 27f9cb55fd575057e77b5738086d36e7a187a03a Mon Sep 17 00:00:00 2001 From: yangdsh Date: Thu, 21 Nov 2019 05:34:54 +0000 Subject: [PATCH] fix bug in simulation.py --- server/analysis/simulation.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/server/analysis/simulation.py b/server/analysis/simulation.py index 4003129..ec08192 100644 --- a/server/analysis/simulation.py +++ b/server/analysis/simulation.py @@ -263,15 +263,15 @@ def run_optimize(X, y, X_samples, model_name, opt_kwargs, model_kwargs): m = gpr_models.create_model(model_name, X=X, y=y, **model_kwargs) timer.stop() model_creation_sec = timer.elapsed_seconds - LOG.info(m._model.as_pandas_table()) + LOG.info(m.model.as_pandas_table()) # Optimize the DBMS's configuration knobs timer.start() - X_news, ypreds, _, _ = tf_optimize(m.model, X_samples, **opt_kwargs) + X_news, _, _, loss = tf_optimize(m.model, X_samples, **opt_kwargs) timer.stop() config_optimize_sec = timer.elapsed_seconds - return X_news, ypreds, m.get_model_parameters(), m.get_hyperparameters() + return X_news, loss, m.get_model_parameters(), m.get_hyperparameters() def gpr_new(env, config, n_loops=100):