def test_get_params(): """Assert that get_params returns the exact dictionary of parameters used by TPOT""" kwargs = { 'population_size': 500, 'generations': 1000, 'config_dict': 'TPOT light', 'offspring_size': 2000, 'verbosity': 1 } tpot_obj = TPOTClassifier(**kwargs) # Get default parameters of TPOT and merge with our specified parameters initializer = inspect.getargspec(TPOTBase.__init__) default_kwargs = dict(zip(initializer.args[1:], initializer.defaults)) default_kwargs.update(kwargs) # update to dictionary instead of input string default_kwargs.update({'config_dict': classifier_config_dict_light}) assert tpot_obj.get_params()['config_dict'] == default_kwargs['config_dict'] assert tpot_obj.get_params() == default_kwargs
def test_get_params(): """Assert that get_params returns the exact dictionary of parameters used by TPOT""" kwargs = {'population_size': 500, 'generations': 1000, 'verbosity': 1} tpot_obj = TPOTClassifier(**kwargs) # Get default parameters of TPOT and merge with our specified parameters initializer = inspect.getargspec(TPOTBase.__init__) default_kwargs = dict(zip(initializer.args[1:], initializer.defaults)) default_kwargs.update(kwargs) assert tpot_obj.get_params() == default_kwargs
def test_get_params(): """Assert that get_params returns the exact dictionary of parameters used by TPOT""" kwargs = { 'population_size': 500, 'generations': 1000, 'verbosity': 1 } tpot_obj = TPOTClassifier(**kwargs) # Get default parameters of TPOT and merge with our specified parameters initializer = inspect.getargspec(TPOTBase.__init__) default_kwargs = dict(zip(initializer.args[1:], initializer.defaults)) default_kwargs.update(kwargs) assert tpot_obj.get_params() == default_kwargs