intial commit (forked from private repo)
This commit is contained in:
12
extracted_cells/cell14.py
Normal file
12
extracted_cells/cell14.py
Normal file
@ -0,0 +1,12 @@
|
||||
# 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()
|
||||
Reference in New Issue
Block a user