Skip to content

A patch for config_dict parameter #431

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

A patch for config_dict parameter #431

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a79079b
fix bug in config dict
Apr 28, 2017
fa1acfb
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
Next Next commit
fix bug in config dict
  • Loading branch information
Weixuan Fu committed Apr 28, 2017
commit a79079b26bb001e5fea98f648eefab04ba4a1eee
14 changes: 10 additions & 4 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 @@ -195,7 +199,9 @@ def __init__(self, generations=100, population_size=100, offspring_size=None,
self.offspring_size = population_size

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 All @@ -209,8 +215,8 @@ def __init__(self, generations=100, population_size=100, offspring_size=None,
else:
try:
with open(config_dict, 'r') as input_file:
file_string = input_file.read()
operator_dict = eval(file_string[file_string.find('{'):(file_string.rfind('}') + 1)])
file_string = input_file.read()
self.config_dict = eval(file_string[file_string.find('{'):(file_string.rfind('}') + 1)])
except:
raise TypeError('The operator configuration file is in a bad format or not available. '
'Please check the configuration file before running TPOT.')
Expand Down