python使用LightGBM预测数据的简单代码实例
时间: 2023-04-01 09:00:46 浏览: 178
功率预测竞赛《Python-通过lightGBM模型进行风电预测》+源代码+实验报告+数据集+文档说明
以下是使用 LightGBM 预测数据的简单 Python 代码实例:
```python
import lightgbm as lgb
import pandas as pd
# 读取数据
train_data = pd.read_csv('train.csv')
test_data = pd.read_csv('test.csv')
# 准备数据
X_train = train_data.drop('target', axis=1)
y_train = train_data['target']
X_test = test_data.drop('target', axis=1)
# 定义模型
params = {
'boosting_type': 'gbdt',
'objective': 'binary',
'metric': 'auc',
'num_leaves': 31,
'learning_rate': .05,
'feature_fraction': .9
}
model = lgb.train(params, lgb.Dataset(X_train, label=y_train))
# 预测
y_pred = model.predict(X_test)
```
以上是一个简单的 LightGBM 预测数据的 Python 代码实例。
阅读全文