首頁 > 軟體

Python手寫回歸樹的實現

2022-06-01 18:00:43

在本篇文章中,我們將介紹迴歸樹及其基本數學原理,並從頭開始使用Python實現一個完整的迴歸樹模型。

為了簡單起見這裡將使用遞回來建立樹節點,雖然遞迴不是一個完美的實現,但是對於解釋原理他是最直觀的。

首先匯入庫

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

首先需要建立訓練資料,我們的資料將具有獨立變數(x)和一個相關的變數(y),並使用numpy在相關值中新增高斯噪聲,可以用數學表達為

這裡的


IT145.com E-mail:sddin#qq.com