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]

参考:

上一篇:4_鸢尾花(k临近)


下一篇:【lssvm预测】基于飞蛾扑火算法改进的最小二乘支持向量机lssvm预测matlab源码