ML之DT：基於簡單迴歸問題訓練決策樹(DIY資料集+三種深度的二元DT效能比較)

輸出結果

設計思路

核心程式碼

for i in range(1, len(xPlot)):
    lhList = list(xPlot[0:i])
    rhList = list(xPlot[i:len(xPlot)])

    lhAvg = sum(lhList) / len(lhList)
    rhAvg = sum(rhList) / len(rhList)

    lhSse = sum([(s - lhAvg) * (s - lhAvg) for s in lhList])
    rhSse = sum([(s - rhAvg) * (s - rhAvg) for s in rhList])

    sse.append(lhSse + rhSse)
    xMin.append(max(lhList))