grid_search.py 文件源码

def svc_model(self, X, y):
        X, y = shuffle(X, y, random_state=1337)
        svc = SVC(kernel='rbf', cache_size=self.cache_size, verbose=True)
        clf = RandomizedSearchCV(svc, param_distributions=self.params, n_iter=self.iters, n_jobs=-1, verbose=self.verbose)
        model = clf.fit(X[0:self.sample_size], y[0:self.sample_size])
        logging.info('Grid Scores ' + str(model.best_params_))
        logging.info('Best Scores ' + str(model.best_score_))
        return model.best_estimator_