LightGBM回归预测代码
时间: 2023-09-09 08:07:11 浏览: 234
lightGBM回归预测代码
5星 · 资源好评率100%
可以直接使用LightGBM库中的API进行回归预测。比如可以用以下代码进行预测:
```python
import lightgbm as lgb
import numpy as np
# 加载模型
bst = lgb.Booster(model_file='model.txt')
# 加载测试数据
test_data = np.array([[1,2,3],[4,5,6]])
# 进行预测
predict = bst.predict(test_data)
print(predict)
```
当然,具体的使用方法还需要根据你的数据及模型配置做出相应的调整。
阅读全文