Skip to content

Restore support for python dictionary in config_dict parameter #430

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
rhiever merged 2 commits into from
Apr 28, 2017
Merged

Restore support for python dictionary in config_dict parameter #430

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
7e4e827
add back support for python dict in config_dict parameter
Apr 28, 2017
66f8ba1
Update base.py
rhiever Apr 28, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 8 additions & 2 deletions tpot/base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -144,7 +144,11 @@ def __init__(self, generations=100, population_size=100, offspring_size=None,
Random number generator seed for TPOT. Use this to make sure
that TPOT will give you the same results each time you run it
against the same data set with that seed.
config_dict: string (default: None)
config_dict: a Python dictionary or string (default: None)
Python dictionary:
A dictionary customizing the operators and parameters that
TPOT uses in the optimization process.
For examples, see config_regressor.py and config_classifier.py
Path for configuration file:
A path to a configuration file for customizing the operators and parameters that
TPOT uses in the optimization process.
Expand Down Expand Up @@ -277,7 +281,9 @@ def __init__(self, generations=100, population_size=100, offspring_size=None,

def _setup_config(self, config_dict):
if config_dict:
if config_dict == 'TPOT light':
if isinstance(config_dict, dict):
self.config_dict = config_dict
elif config_dict == 'TPOT light':
if self.classification:
self.config_dict = classifier_config_dict_light
else:
Expand Down