optimize_model_ensemble.py 文件源码

def create_voting_classifier_ensemble(model_tuple_list):
    '''
    INPUT
         - model tuple list: list of model tuples (name, model)
    OUTPUT
         - a fit ensemble

    Return fit voting ensemble
    '''
    ensemble = VotingClassifier(model_tuple_list, voting='soft')
    ensemble.fit(X_train_b, y_train_b)
    return ensemble