Beräkna särdragsvikt med random forests
Som datascientist på ett finansiellt konsultföretag har du utvecklat en random forest-klassificerare som klassificerar individer utifrån deras inkomstnivåer. Nu behöver du förklara modellen genom att analysera särdragsvikterna för att identifiera de viktigaste faktorerna för att förutsäga inkomst – något som möjliggör mer träffsäker marknadssegmentering och förbättrat strategiskt beslutsfattande.
matplotlib.pyplot har importerats som plt. X_train och y_train är förinlästa.
Den här övningen är en del av kursen
Förklarbar AI i Python
Övningsinstruktioner
- Extrahera särdragsvikterna från
model. - Plotta
feature_importances.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
model = RandomForestClassifier(random_state=42)
model.fit(X_train, y_train)
# Derive feature importances
feature_importances = ____
# Plot the feature importances
____
plt.show()