Last active
August 16, 2018 11:22
-
-
Save jpmallette/6fc836861b3539d6b2dc09469a947c1f to your computer and use it in GitHub Desktop.
minimalist Parameter Grid
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from itertools import product | |
| from fbprophet import Prophet | |
| m = Prophet() | |
| param_grid = { | |
| 'model' : [m] | |
| , 'initial' : ['730 days','500 days'] | |
| , 'period' : ['180 days'] | |
| , 'horizon' : ['365 days'] | |
| } | |
| def create_grid(param_grid): | |
| param_grid_list = [param_grid].copy() | |
| for p in param_grid_list: | |
| # Always sort the keys of a dictionary, for reproducibility | |
| items = sorted(p.items()) | |
| if not items: | |
| yield {} | |
| else: | |
| keys, values = zip(*items) | |
| for v in product(*values): | |
| params = dict(zip(keys, v)) | |
| yield params | |
| >>> list(create_grid(param_grid)) | |
| [{'model': <fbprophet.forecaster.Prophet at 0x11cd0bba8>, | |
| 'initial': '730 days', | |
| 'period': '180 days', | |
| 'horizon': '365 days'}, | |
| {'model': <fbprophet.forecaster.Prophet at 0x11cd0bba8>, | |
| 'initial': '500 days', | |
| 'period': '180 days', | |
| 'horizon': '365 days'}] |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment