Implementasi Random Forest – Part 4

Pada modul ini kita akan menampilkan insight yang dihasilkan oleh model.

#Feature Importance
importance = pd.Series(model.feature_importances_,
                       index = X_train.columns.values)
importance.nlargest(5).plot(kind = 'barh',
                            figsize = (9,6))

Dapat dilihat, feature expenditure, share, reports, age dan active adalah fitur penting