Berechnung des Einflusses von Merkmalen mit logistischer Regression
Du setzt deine Arbeit beim Versicherungsunternehmen fort und hast ein prädiktives Modell entwickelt, um vorherzusagen, ob jemand raucht oder nicht. Jetzt musst du das Modell analysieren, um die relevanten Faktoren zu bestimmen, anhand derer jemand als Raucher oder Nichtraucher kategorisiert wird. So kann das Unternehmen das Risiko genauer einschätzen und die Versicherungspolicen entsprechend anpassen.
matplotlib.pyplot
wurde als plt
importiert. X_train
und y_train
sind bereits geladen.
Diese Übung ist Teil des Kurses
Erklärbare KI in Python
Anleitung zur Übung
- Extrahiere die Koeffizienten (
coefficients
) aus dem Modell. - Erstelle ein Diagramm der Koeffizienten für die angegebenen Merkmalsnamen (
feature_names
).
Interaktive Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
scaler = MinMaxScaler()
X_train_scaled = scaler.fit_transform(X_train)
model = LogisticRegression()
model.fit(X_train_scaled, y_train)
# Derive coefficients
coefficients = ____
feature_names = X_train.columns
# Plot coefficients
____
plt.show()