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

• 【機器學習】多項式迴歸原理介紹
• 【機器學習】多項式迴歸python實現
• 【機器學習】多項式迴歸sklearn實現

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

使用一個簡單的數據集來模擬，只有幾條數據。

代碼

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

運行結果

從圖中看出數據分佈在一條拋物線附近。

最終獲得的模型爲：

若是須要代碼和數據集，請掃描下面二維碼關注公衆號【AI developer】，回覆【代碼】便可免費獲取。