這裡使用sklearn框架實現線性迴歸。使用框架更方便,可以少寫很多程式碼。
寫了三個例子,分別是單變數的、雙變數的和多變數的。單變數和雙變數的畫出了圖,多變數的由於高維空間難以實現,所以沒有畫圖。單變數和雙變數的使用的自己模擬的一個簡單的房價資料集,多變數的使用的boston房價資料集。
1.單變數線性迴歸
程式碼

執行結果


2.雙變數線性迴歸
程式碼

執行結果


3.多變數線性迴歸
程式碼

執行結果

如果需要程式碼和資料集,請掃描下面二維碼關注公眾號【AI developer】,回覆【程式碼】即可免費獲取。
