Computing feature impact with linear regression
As a data scientist at an insurance company, your task is to build and explain a linear regression model that estimates insurance charges based on features like age, BMI, and smoking status by analyzing the model's coefficients to determine the impact of each feature on the predictions.
matplotlib.pyplot has been imported as plt along with MinMaxScaler. X_train and y_train are pre-loaded for you.
Deze oefening maakt deel uit van de cursus
Explainable AI in Python
Oefeninstructies
- Normalize the training data
X_train. - Fit the linear regression
modelto the standardized training data. - Extract the
coefficientsfrom the model. - Plot the
coefficientsfor the givenfeature_names.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# 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()