1. 程式人生 > >EL之Boosting之GB:利用梯度提升法解決迴歸(對多變數的資料集+實數值評分預測)問題

EL之Boosting之GB:利用梯度提升法解決迴歸(對多變數的資料集+實數值評分預測)問題

EL之Boosting之GB:利用梯度提升法解決迴歸(對多變數的資料集+實數值評分預測)問題

輸出結果

 

設計思路

 

核心程式碼

xList = []
labels = []
names = []
firstLine = True
for line in data:
    if firstLine:
        names = line.decode().strip().split(";")
        firstLine = False
    else:
        row = line.decode().strip().split(";")
        labels.append(float(row[-1]))
        row.pop()
        floatRow = [float(num) for num in row]
        xList.append(floatRow)