LightGBM简单例子

安装

pip install lightgbm
conda install lightgbm

【python】conda和pip安装库之间的区别

代码

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://lightgbm.readthedocs.io/en/latest/Python-Intro.html