LGBM 模型保存为PMML 文件

本文介绍了如何使用LightGBM训练模型并将其保存为PMML文件。首先,定义了模型参数,包括任务类型、提升方法、目标函数等,并进行了训练。接着,模型被保存为txt文件。然后,下载了将模型转换为PMML文件所需的jar包。最后,通过在命令行中运行指定的java命令,将模型从txt文件转换为PMML格式。
摘要由CSDN通过智能技术生成

1、训练模型

import lightgbm as lgb
params = {
    'task': 'train',
    'boosting_type': 'gbdt',  # 设置提升类型
    'objective': 'binary',  # 目标函数  regression
    'metric': {'auc'},  # 评估函数   metric = binary_logloss,auc

    'learning_rate': 0.05,  # [0.01, 0.02, 0.05, 0.1, 0.15]
    'n_estimators': 500,   # 默认是100.也就是循环次数,或者叫树的数目。

    'max_depth': 3,  # range(5, 15, 2)
    'num_leaves': 16,  # 一棵树上的叶子数 [15, 25, 40, 55]
    'min_child_samples': 300,  # 一个叶子上数据的最小数量
    'min_child_weight': 10,  # 分支结点的最小权重
    'feature_fraction': 0.7,  # 每次新建一棵树时,随机使用多少的特征。
    'bagging_fraction': 0.7,
    'bagging_freq': 10,  # bagging 的频率
    # 'cat_smooth': 30,  # 这可以降低噪声在分类特征中的影响
    'lambda_l1': 4,  # reg_

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值