enron_poi_ml_ci.py 文件源码

def opt_classifier(clf, params, features_train, labels_train, optimize=True):
    '''
    GridSearchCV to find optimal parameters of the classifier.
    '''

    if optimize:
        scorer = make_scorer(f1_score)
        clf = GridSearchCV(clf, params, scoring=scorer)
        clf = clf.fit(features_train, labels_train)
        clf = clf.best_estimator_
    else:
        clf = clf.fit(features_train, labels_train)

    return clf