天天看點

EL之GB(GBR):利用GBR對回歸問題(實數值評分預測)模組化

輸出結果

EL之GB(GBR):利用GBR對回歸問題(實數值評分預測)模組化
EL之GB(GBR):利用GBR對回歸問題(實數值評分預測)模組化

設計思路

EL之GB(GBR):利用GBR對回歸問題(實數值評分預測)模組化

核心代碼

wineGBMModel = ensemble.GradientBoostingRegressor(n_estimators=nEst,

                                                 max_depth=depth,

                                                 learning_rate=learnRate,

                                                 subsample = subSamp,

                                                 loss='ls')

wineGBMModel.fit(xTrain, yTrain)

msError = []

predictions = wineGBMModel.staged_predict(xTest)

繼續閱讀