机器学习：多个分类模型建模融合

最新推荐文章于 2025-04-24 11:32:17 发布

rubyw

最新推荐文章于 2025-04-24 11:32:17 发布

阅读量208

点赞数 3

CC 4.0 BY-SA版权

分类专栏：机器学习金融风险分析文章标签：机器学习分类数据分析 python

本文链接：https://blog.youkuaiyun.com/rubyw/article/details/141125698

def cv_model(clf, train_x, train_y, test_x, clf_name):
    folds = 10
    seed = 2020
    kf = KFold(n_splits=folds, shuffle=True, random_state=seed)

    train = np.zeros(train_x.shape[0])
    test = np.zeros(test_x.shape[0])

    cv_scores = []

    for i, (train_index, valid_index) in enumerate(kf.split(train_x, train_y)):
        print('************************************ {} ************************************'.format(str(i+1)))
        trn_x, trn_y, val_x, val_y = train_x.iloc[train_index], train_y[train_index], train_x.iloc[valid_index], train_y[valid_index]

        if clf_name == "lgb":
            train_matrix = clf.Dataset(trn_x, label=trn_y)
            valid_matrix = clf.Dataset(val_x, label=val_y)

            params = {
   
                'boosting_type': 'gbdt'