unExtGBDTEnsemble.py 文件源码

def ExtGBDTEnsemblePredict(sub_clf_num, predict_x):
    """
    ????????
    :param sub_clf_num: ??????
    :param predict_x: ??????feature
    :return: socre: ndarray, ????
    """
    total_score = np.zeros(len(predict_x))  # ?????????????????
    for i in range(sub_clf_num):
        predict_X = xgb.DMatrix(predict_x)
        model_file = '../model/model' + str(i)
        bst = pickle.load(open(model_file, 'r'))
        predict_y = bst.predict(predict_X)
        total_score += predict_y
    score = total_score / sub_clf_num
    return score