HyperOptSearch.py 文件源码

def start_search_server(self, mongo_uri, data_filename, layers_num, max_evals=100, nb_epoch = 10, patience = 5):
        space = self.conv_model_space(layers_num, data_filename)
        exp_key, trials = self.create_mongo_trials(mongo_uri)
        keras_model = KerasModel(exp_key, nb_epoch = nb_epoch, patience = patience, mongo_uri = mongo_uri)
        best = fmin(keras_model.run_job, space, trials=trials, algo=tpe.suggest, max_evals=max_evals)
        print "Done!"
        print best