# Select only the 'persons' and 'safety' columns from X_train and X_test
X_train_selected = X_train[['persons', 'safety']]
X_test_selected = X_test[['persons', 'safety']]
print("Shape of X_train_selected:", X_train_selected.shape)
print("Shape of X_test_selected:", X_test_selected.shape)
display(X_train_selected.head())
import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
importances = tree.feature_importances_
columns = X.columns
# Create a Series for feature importances and filter for non-zero values
feature_importance_series = pd.Series(importances, index=columns)
important_features = feature_importance_series[feature_importance_series > 0]
# Sort for better visualization
important_features = important_features.sort_values(ascending=False)
sns.barplot(x=important_features.index, y=important_features.values, palette = 'bright', saturation = 2.0, edgecolor ='black', linewidth = 2)
plt.title('Importancia de cada Feature (Top Importantes)')
plt.xlabel('Features')
plt.ylabel('Importance Score')
plt.show()