AttributeError: 'LGBMClassifier' object has no attribute 'feature_importance'.
时间: 2024-01-25 18:13:20 浏览: 65
在LightGBM中,如果出现了'AttributeError: 'LGBMClassifier' object has no attribute 'feature_importance''的错误,这意味着LGBMClassifier对象没有feature_importance属性。在LightGBM中,要获取特征的重要性排序,可以使用`feature_importances_`属性。
以下是一个示例代码,演示如何使用LightGBM获取特征的重要性排序:
```python
import lightgbm as lgb
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
# 加载数据集
data = load_breast_cancer()
X = data.data
y = data.target
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建LightGBM分类器
clf = lgb.LGBMClassifier()
# 训练模型
clf.fit(X_train, y_train)
# 获取特征的重要性排序
feature_importances = clf.feature_importances_
# 打印特征的重要性排序
for feature_name, importance in zip(data.feature_names, feature_importances):
print(f"{feature_name}: {importance}")
```
这段代码使用了LightGBM库中的LGBMClassifier类来训练一个分类模型,并使用`feature_importances_`属性获取特征的重要性排序。你可以根据自己的数据集和需求进行相应的修改。