CommencerCommencer gratuitement

Calculer l’impact des variables avec une régression logistique

Dans la continuité de votre travail au sein de la compagnie d’assurance, vous avez construit un modèle prédictif pour identifier si une personne est fumeuse ou non. Vous devez à présent analyser le modèle afin de déterminer les facteurs pertinents qui influencent le statut de fumeur, afin d’aider l’entreprise à évaluer le risque plus précisément et à adapter ses contrats en conséquence.

matplotlib.pyplot a été importé sous le nom plt. X_train et y_train sont préchargés pour vous.

Cet exercice fait partie du cours

IA explicable en Python

Afficher le cours

Instructions

  • Extrayez les coefficients du modèle.
  • Tracez les coefficients pour les feature_names fournis.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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()
Modifier et exécuter le code