samplers.py 文件源码

def __init__(self, experiment, args, job_module_config):
        super(self.__class__, self).__init__(experiment, args, job_module_config)
        # pre-format the experiment dict
        # Sklearn needs all the params to be in a  list for the grid to work
        # properly
        for param in experiment['params']:
            if type(experiment['params'][param]) is not list:
                experiment['params'][param] = [experiment['params'][param] ]

        self.searcher = ParameterGrid(experiment['params'])