利用 test_data = lgb.Dataset(X_test, label=y_test)进行测试
时间: 2024-01-24 19:17:37 浏览: 124
lightGBM回归预测代码
5星 · 资源好评率100%
是的,使用`test_data`数据集进行测试,可以通过LightGBM模型的`predict()`方法来进行预测。具体的代码如下:
```python
# 定义测试数据集
test_data = lgb.Dataset(X_test, label=y_test)
# 使用测试数据集进行预测
y_pred = model.predict(X_test)
# 计算预测的准确率
accuracy = accuracy_score(y_test, np.round(y_pred))
print('Accuracy:', accuracy)
```
其中,`model`是训练好的LightGBM模型。`np.round()`函数将预测值四舍五入为最接近的整数,因为LightGBM默认输出的是浮点数。`accuracy_score()`函数计算预测的准确率。
阅读全文