ÎncepețiÎncepe gratuit

Calcularea impactului caracteristicilor cu regresia liniară

În calitate de om de știință a datelor la o companie de asigurări, sarcina ta este să construiești și să explici un model de regresie liniară care estimează costurile asigurărilor pe baza unor caracteristici precum vârsta, IMC-ul și statutul de fumător. Vei analiza coeficienții modelului pentru a determina impactul fiecărei caracteristici asupra predicțiilor.

matplotlib.pyplot a fost importat ca plt, împreună cu MinMaxScaler. X_train și y_train sunt preîncărcate pentru tine.

Acest exercițiu face parte din cursul

AI Explicabil în Python

Vezi cursul

Instrucțiuni pentru exercițiu

  • Normalizează datele de antrenament X_train.
  • Antrenează modelul de regresie liniară model pe datele de antrenament standardizate.
  • Extrage coefficients din model.
  • Reprezintă grafic coefficients pentru feature_names date.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Standardize the training data
scaler = MinMaxScaler()
X_train_scaled = ____
model = LinearRegression()

# Fit the model
____

# Derive coefficients
coefficients = ____
feature_names = X_train.columns

# Plot coefficients
plt.bar(____, ____)
plt.show()
Editează și rulează codul