XGBoost mendapatkan fitur penting sebagai daftar kolom, bukan plot

Saya ingin tahu apakah Anda bisa mendapatkan fitur penting sebagai daftar kolom, bukan plot. Inilah yang saya punya

xg_reg = xgb.train(params=params, dtrain=data_dmatrix, num_boost_round=10)
import matplotlib.pyplot as plt

xgb.plot_importance(xg_reg)
plt.rcParams['figure.figsize'] = [5,5]
plt.show()

Yang memberi saya plot ini

masukkan deskripsi gambar di sini

Saya hanya ingin mendapatkan daftar fitur teratas karena saya memiliki lebih dari 800 fitur berbeda.


person Snorrlaxxx    schedule 23.07.2020    source sumber


Jawaban (1)


Anda dapat menggunakan xgb.get_score(). Berikut beberapa contohnya: Bagaimana cara mendapatkan fitur penting di xgboost?

person jared_mamrot    schedule 24.07.2020