Mulai sekarangMulai gratis

Plot Feature Importance untuk analisis penerimaan

Sebagai bagian dari tim data science di sebuah universitas, tugas Anda adalah mengevaluasi faktor-faktor yang benar-benar mendorong keputusan penerimaan dan faktor mana yang kurang penting. Meskipun komite penerimaan mengetahui bahwa CGPA berperan penting, mereka ingin mengonfirmasinya dan menemukan faktor penting lain yang mungkin memengaruhi hasil. Dengan menggunakan model RandomForestRegressor, Anda akan memvisualisasikan feature importance untuk mengidentifikasi dengan jelas aspek profil pelamar mana yang paling berpengaruh dan mana yang berdampak lebih kecil pada proses pengambilan keputusan.

Library shap dan data pelatihan (X_train, y_train) telah dimuat untuk Anda.

Latihan ini merupakan bagian dari kursus

Explainable AI in Python

Lihat Kursus

Instruksi latihan

  • Turunkan shap_values menggunakan TreeExplainer.
  • Gunakan shap_values yang diperoleh untuk membuat bar plot feature importance dan analisislah hasilnya.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

model = RandomForestRegressor(random_state=42)
model.fit(X_train, y_train)

# Derive shap values
explainer = ____
shap_values = ____

# Plot the feature importance plot
____
Edit dan Jalankan Kode