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
Instruksi latihan
- Turunkan
shap_valuesmenggunakanTreeExplainer. - Gunakan
shap_valuesyang 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
____