阅读 68

LightGBM简单例子

安装

pip install lightgbm
conda install lightgbm

代码

from lightgbm import LGBMClassifier

X = [[1,2],[3,4],[5,6],[7,8],[9,10]]
y = [0,0,0,1,1]
model = LGBMClassifier()
model.fit(X,y)
print(model.predict([[5,5]]))

代码中的X是特征变量,共有2个特征;
代码中的y是目标变量;
然后代码引入模型;
代码用fit()函数训练模型;
代码用predict()函数进行预测,预测结果如下。

[0]

参考:

原文:https://www.cnblogs.com/ghj1976/p/lightgbm-jian-dan-li-zi.html

文章分类
代码人生
文章标签
版权声明:本站是系统测试站点,无实际运营。本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 XXXXXXo@163.com 举报,一经查实,本站将立刻删除。
相关推荐