请写一个lgbm算法的代码
时间: 2024-04-12 10:29:25 浏览: 159
lightgbmm_lgbm_lightGBM_
5星 · 资源好评率100%
当然可以!下面是一个简单的示例代码,使用LightGBM(LGBM)算法进行分类任务:
```python
import lightgbm as lgb
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# 准备数据集
X, y = load_data() # 加载你的数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建LightGBM数据集
train_data = lgb.Dataset(X_train, label=y_train)
# 设置参数
params = {
'objective': 'binary',
'metric': 'binary_logloss',
'boosting_type': 'gbdt',
'num_leaves': 31,
'learning_rate': 0.05,
'feature_fraction': 0.9,
'bagging_fraction': 0.8,
'bagging_freq': 5,
'verbose': 0
}
# 训练模型
model = lgb.train(params, train_data, num_boost_round=100)
# 预测
y_pred = model.predict(X_test)
y_pred_binary = [1 if p >= 0.5 else 0 for p in y_pred]
# 评估模型
accuracy = accuracy_score(y_test, y_pred_binary)
print(f"Accuracy: {accuracy}")
```
请注意,这只是一个简单的示例代码,你需要根据你的实际需求进行修改和调整。如果你具体的数据集和任务要求,请提供更多细节,我可以给出更加具体的代码帮助。
阅读全文