Bewertung von Auswirkungen mit Diagrammen zur partiellen Abhängigkeit
Aufbauend auf deinen vorherigen Arbeiten wirst du nun untersuchen, wie der Notendurchschnitt ('CGPA') und die Hochschulbewertung ('University Rating') die Zulassungsentscheidungen beeinflussen. Unsere frühere Analyse hat gezeigt, dass der Notendurchschnitt der wichtigste Prädiktor ist, während die Hochschulbewertung den geringsten Stellenwert hat. Mithilfe eines Diagramms zur partiellen Abhängigkeit können wir sehen, wie sich Veränderungen dieser Merkmale auf die Zulassungswahrscheinlichkeit auswirken.
X_train
und y_train
wurden bereits geladen.
Diese Übung ist Teil des Kurses
Erklärbare KI in Python
Interaktive Übung zum Anfassen
Probieren Sie diese Übung aus, indem Sie diesen Beispielcode ausführen.
from sklearn.ensemble import RandomForestRegressor
import shap
model = RandomForestRegressor(random_state=42)
model.fit(X_train, y_train)
# Generate the partial dependence plot for CGPA
shap.partial_dependence_plot(____, ____, X_train)