1. 程式人生 > >EL之RF:利用RandomForestRegressor對迴歸問題(實數值評分預測)建模(調2參)

EL之RF:利用RandomForestRegressor對迴歸問題(實數值評分預測)建模(調2參)

EL之RF:利用RandomForestRegressor對迴歸問題(實數值評分預測)建模(調2參)

輸出結果

 

設計思路

 

核心程式碼

mseOos = []
nTreeList = range(100, 1000, 100)    #----▲☆▲☆▲
for iTrees in nTreeList:
#     depth = None
#     depth=6      #----▲▲▲▲▲
    depth=10     #----☆☆☆☆
    
    maxFeat  = 4 #try tweaking

    wineRFModel = ensemble.RandomForestRegressor(n_estimators=iTrees, max_depth=depth, max_features=maxFeat,
                                                 oob_score=False, random_state=531)
    
    wineRFModel.fit(xTrain,yTrain)
    prediction = wineRFModel.predict(xTest)
    mseOos.append(mean_squared_error(yTest, prediction))