視覺化線性迴歸模型
現在你已經建立線性迴歸模型,並用所有可用的觀測資料完成訓練,可以來視覺化模型對資料的擬合情形。這能幫助你解讀「radio 廣告支出」與「sales 銷售額」之間的關係。
變數 X(radio 數值的陣列)、y(sales 數值的陣列),以及 predictions(在給定 X 時,模型對 y 的預測值陣列)都已替你從上一個練習預先載入。
本練習屬於課程
使用 scikit-learn 進行監督式學習
練習說明
- 導入
matplotlib.pyplot並命名為plt。 - 建立一個散佈圖,將
y對X視覺化,點的顏色為藍色。 - 繪製一條紅色折線圖,顯示
predictions對X的關係。 - 顯示圖表。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Import matplotlib.pyplot
import ____.____ as ____
# Create scatter plot
plt.scatter(____, ____, color="____")
# Create line plot
plt.plot(____, ____, color="____")
plt.xlabel("Radio Expenditure ($)")
plt.ylabel("Sales ($)")
# Display the plot
plt.____()