【機器學習】多項式迴歸python實現

AI_developer發表於2019-03-10

使用python實現多項式迴歸,沒有使用sklearn等機器學習框架,目的是幫助理解演算法的原理。

使用一個簡單的資料集來模擬,只有幾條資料。

【機器學習】多項式迴歸python實現

程式碼

【機器學習】多項式迴歸python實現

  1. 從資料集中讀取X和y。
  2. 為X新增二次方項,用Z替換。
  3. 給Z新增 1 列,初始化為 1 ,用來求偏置項。
  4. 劃分訓練集和測試集。
  5. 將Z和y的訓練集轉換為矩陣形式。
  6. 和線性迴歸類似,使用正規方程法,先驗證矩陣的可逆性。
  7. 去掉Z中全為1的列。
  8. 使用測試集驗證模型。
  9. Z和y的訓練集轉換回陣列形式。
  10. 列印結果和圖片。

執行結果

在這裡插入圖片描述
從圖中看出資料分佈在一條拋物線附近。

【機器學習】多項式迴歸python實現
最終得到的模型為:

【機器學習】多項式迴歸python實現
如果需要程式碼和資料集,請掃描下面二維碼關注公眾號【AI developer】,回覆【程式碼】即可免費獲取。

【機器學習】多項式迴歸python實現

相關文章