lightgbm 训练模型代码

最新推荐文章于 2025-05-27 18:00:51 发布

国企打工人

最新推荐文章于 2025-05-27 18:00:51 发布

阅读量792

点赞数

CC 4.0 BY-SA版权

分类专栏：机器学习算法文章标签： python

本文链接：https://blog.youkuaiyun.com/tongjinrui/article/details/120854841

机器学习算法专栏收录该内容

19 篇文章

订阅专栏

该博客介绍了如何使用LightGBM库训练一个二分类模型，详细配置了参数，包括boosting_type、num_leaves、learning_rate等，并在训练过程中应用了早停策略。最终通过ROC_AUC分数评估模型性能，并导出了特征重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import lightgbm as lgb
import pandas as pd


def lgb_train(x_train, y_train, x_test, y_test):
    clf = lgb.LGBMClassifier(
        boosting_type='gbdt',
        # num_leaves=31,
        num_leaves=31,
        reg_alpha=0.05,
        reg_lambda=1.5,
        max_depth=-1,
        n_estimators=1200,
        objective='binary',
        subsample=0.85,
        colsample_bytree=0.8,
        subsample_freq=1,
        learning_rate=0.035,
#         min_child_weight=0.1,
        min_child_samples=25,
        seed=2021,
        nthread=-1)

    clf.fit(X, y, eval_metric='auc', verbose=True, eval_set=[(x_train,y_train),(x_test,y_test)], early_stopping_rounds=5,categorical_feature='auto')
    
    y_pred = clf.predict_proba(x_test)[:, 1]
    auc = roc_auc_score(y_test, y_pred)
    print(f"lgb-auc:  {auc}")
    importance = pd.Series(data=clf.feature_importances_, index=feature_col, name=("colname", "importance")).sort_values(
        ascending=False).to_frame()
    importance.to_csv('import.csv')
    return clf