Skip to content

Instantly share code, notes, and snippets.

@gumdropsteve

gumdropsteve/set_param_combos.py Secret

Created October 1, 2020 07:44
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save gumdropsteve/8035c30081f4724cb564c94e100298a0 to your computer and use it in GitHub Desktop.
Save gumdropsteve/8035c30081f4724cb564c94e100298a0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
set_param_combos.py
import itertools
param_grid = {'changepoint_prior_scale': [0.001, 0.01, 0.1, 0.5],
'seasonality_prior_scale': [0.01, 0.1, 1.0, 10.0],
'seasonality_mode': ['additive', 'multiplicative']}
# generate all combinations of parameters
all_params = [dict(zip(param_grid.keys(), v)) for v in itertools.product(*param_grid.values())]
rmses = [] # store the RMSEs for each params here
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment