開始使用免費開始

用線性迴歸計算特徵影響力

身為一家保險公司的資料科學家,你的任務是建立並說明一個線性迴歸模型,根據年齡、BMI、是否吸菸等特徵來估計保險費用,並透過分析模型的係數來判斷各個特徵對預測的影響程度。

matplotlib.pyplot 已以 plt 匯入,MinMaxScaler 亦可使用。X_trainy_train 已為你預先載入。

本練習屬於課程

Python 的 Explainable AI

檢視課程

練習說明

  • 正規化訓練資料 X_train
  • 將線性迴歸 model 擬合到標準化後的訓練資料。
  • 從模型中擷取 coefficients
  • 針對給定的 feature_names 繪製 coefficients

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# 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()
編輯並執行程式碼