python lightgbm算法
时间: 2023-06-16 13:06:50 浏览: 159
temp_python代码_lightGBM_
LightGBM 是一种基于决策树的梯度提升框架,它是 Microsoft 开源的一个高效的机器学习算法库。它的特点是速度快、内存占用低、准确性高、可扩展性强,尤其在处理大规模数据时表现出色。
下面是使用 LightGBM 进行简单的分类模型的示例代码:
```python
import lightgbm as lgb
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# 加载数据集
data = load_breast_cancer()
X_train, X_test, y_train, y_test = train_test_split(data.data, data.target, test_size=0.2, random_state=42)
# 构建 LightGBM 数据集
train_data = lgb.Dataset(X_train, label=y_train)
test_data = lgb.Dataset(X_test, label=y_test)
# 设置模型参数
params = {
'boosting_type': 'gbdt',
'objective': 'binary',
'metric': 'binary_logloss',
'num_leaves': 31,
'learning_rate': 0.05,
'feature_fraction': 0.9
}
# 训练模型
num_rounds = 100
model = lgb.train(params, train_data, num_rounds)
# 预测并评估模型
y_pred = model.predict(X_test)
y_pred = [1 if x >= 0.5 else 0 for x in y_pred]
accuracy = accuracy_score(y_test, y_pred)
print('Accuracy:', accuracy)
```
在这个示例中,我们使用了 LightGBM 提供的 `lgb.Dataset` 类来构建训练集和测试集,然后通过设置 `params` 字典来指定模型的参数,包括使用的算法、评价指标、树的数量等。最后,使用 `lgb.train` 函数训练模型,并使用预测结果计算准确率。
阅读全文