# 데이터 처리
X = x.value
y
feature_names = list(x.column)
# 모델
model = xgb.XGBClassifier()
model.fit(X, y)
# 이름 붙이기
# https://stackoverflow.com/questions/46943314/xgboost-plot-importance-doesnt-show-feature-names
model.get_booster().feature_names = feature_names
xgb.plot_importance(model.get_booster())
# importance
importance = model.feature_importances_
# 저장
importance_df = pd.DataFrame({'Feature': X_train.columns, 'Importance': importance})
importance_df.to_csv('~/xgboost.csv', index=False)
반응형