使用python實現多項式迴歸,沒有使用sklearn等機器學習框架,目的是幫助理解演算法的原理。
使用一個簡單的資料集來模擬,只有幾條資料。
程式碼
- 從資料集中讀取X和y。
- 為X新增二次方項,用Z替換。
- 給Z新增 1 列,初始化為 1 ,用來求偏置項。
- 劃分訓練集和測試集。
- 將Z和y的訓練集轉換為矩陣形式。
- 和線性迴歸類似,使用正規方程法,先驗證矩陣的可逆性。
- 去掉Z中全為1的列。
- 使用測試集驗證模型。
- Z和y的訓練集轉換回陣列形式。
- 列印結果和圖片。
使用python實現多項式迴歸,沒有使用sklearn等機器學習框架,目的是幫助理解演算法的原理。
使用一個簡單的資料集來模擬,只有幾條資料。