使用 regplot() 視覺化模型擬合
在擬合並分析過模型後,你可以透過繪製觀測點與擬合出的羅吉斯迴歸來進行視覺化。
利用這個圖,你可以直觀看出自變數在其取值範圍內與應變數之間的關係。
我們可以使用 seaborn 模組中的 regplot() 函式。regplot() 有一個參數 logistic,你可以用 True 或 False 指定是否要對給定資料估計羅吉斯迴歸模型。這同時也會繪出擬合曲線。
回想你先前擬合的模型:
$$
\log\bigg(\frac{y}{1-y}\bigg) = -0.3055 + 0.3791*\text{arsenic}
$$
資料集 wells 已經載入到你的工作環境中。
本練習屬於課程
Generalized Linear Models in Python
練習說明
- 使用
wells資料,將arsenic繪在 x 軸、switch繪在 y 軸。 - 設定
y_jitter為 0.03,讓應變數的取值稍微分散以利觀察。 - 將參數
logistic設為True,讓圖上疊加羅吉斯函式,並將信賴區間參數ci設為None,不顯示信賴區間,同時加快運算。 - 使用
plt.show()顯示圖表。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Plot arsenic and switch and add overlay with the logistic fit
sns.regplot(x = ____, y = ____,
y_jitter = ____,
data = ____,
logistic = ____,
ci = ____)
# Display the plot
____