安装
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]
参考: