天天看點

機器學習——模型樹

和回歸樹(在每個葉節點上使用各自的均值做預測)不同,模型樹算法需要在每個葉節點上都建構出一個線性模型,這就是把葉節點設定為分段線性函數,這個所謂的分段線性(piecewise linear)是指模型由多個線性片段組成。

main.py

得到兩段函數,以0.28為分界

分别為y=3.46877+1.1852x和y=0.001698+11.96477x

而生成該資料的真實模型是y=3.5+1.0x和y=0+12x再加上高斯噪聲生成

機器學習——模型樹