12 lines
402 B
Python
12 lines
402 B
Python
# Feature Importance
|
|
feature_importance = pd.DataFrame({
|
|
'Feature': X_train.columns,
|
|
'Importance': rf_model.feature_importances_
|
|
}).sort_values('Importance', ascending=False)
|
|
|
|
plt.figure(figsize=(12, 8))
|
|
sns.barplot(x='Importance', y='Feature', data=feature_importance.head(15))
|
|
plt.title('Feature Importance')
|
|
plt.tight_layout()
|
|
plt.savefig('output/Feature_Importance.png', dpi=300)
|
|
plt.show() |