在数据框中获取重要特征

时间:2021-06-23 18:59:25

标签: pandas numpy xgboost

请问如何获取dataframe中的重要特征

# fit model to training data
xgb_model = XGBClassifier(random_state = 0 )
xgb_model.fit(X_train, y_train)

print("Feature Importances : ", xgb_model.feature_importances_) 

我知道如何绘制它,但我想知道如何将 20 个最重要的特征放入数据框或列表中

0 个答案:

没有答案