debug for new gpr

This commit is contained in:
bohanjason 2019-12-05 00:06:37 -05:00 committed by Dana Van Aken
parent 70f9f952d5
commit 1e17c78956
2 changed files with 4 additions and 1 deletions

View File

@ -27,6 +27,8 @@ FLIP_PROB_DECAY = 0.5
# ---GPR CONSTANTS---
USE_GPFLOW = True
GPR_DEBUG = True
DEFAULT_LENGTH_SCALE = 2.0
DEFAULT_MAGNITUDE = 1.0

View File

@ -41,7 +41,7 @@ from website.settings import (USE_GPFLOW, DEFAULT_LENGTH_SCALE, DEFAULT_MAGNITUD
ACTOR_HIDDEN_SIZES, CRITIC_HIDDEN_SIZES,
DNN_TRAIN_ITER, DNN_EXPLORE, DNN_EXPLORE_ITER,
DNN_NOISE_SCALE_BEGIN, DNN_NOISE_SCALE_END,
DNN_DEBUG, DNN_DEBUG_INTERVAL)
DNN_DEBUG, DNN_DEBUG_INTERVAL, GPR_DEBUG)
from website.settings import INIT_FLIP_PROB, FLIP_PROB_DECAY
from website.types import VarType
@ -627,6 +627,7 @@ def configuration_recommendation(recommendation_input):
opt_kwargs['learning_rate'] = DEFAULT_LEARNING_RATE
opt_kwargs['maxiter'] = MAX_ITER
opt_kwargs['bounds'] = [X_min, X_max]
opt_kwargs['debug'] = GPR_DEBUG
ucb_beta = 'get_beta_td'
opt_kwargs['ucb_beta'] = ucb.get_ucb_beta(ucb_beta, scale=DEFAULT_UCB_SCALE,
t=i + 1., ndim=X_scaled.shape[1])