開始使用免費開始

使用 regplot() 視覺化模型擬合

在擬合並分析過模型後,你可以透過繪製觀測點與擬合出的羅吉斯迴歸來進行視覺化。

利用這個圖,你可以直觀看出自變數在其取值範圍內與應變數之間的關係。

我們可以使用 seaborn 模組中的 regplot() 函式。regplot() 有一個參數 logistic,你可以用 TrueFalse 指定是否要對給定資料估計羅吉斯迴歸模型。這同時也會繪出擬合曲線。

回想你先前擬合的模型:
$$ \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
____
編輯並執行程式碼