EL之RF(RFR):利用RF对回归(性别属性编码)问题(整数值年龄预测)建模

EL之RF(RFR):利用RF对回归(性别属性编码)问题(整数值年龄预测)建模


输出结果

设计思路

核心代码

for iTrees in nTreeList:
    depth = None
    abaloneRFModel = ensemble.RandomForestRegressor(n_estimators=iTrees, max_depth=depth, max_features=maxFeat,
                                                 oob_score=False, random_state=531)

    abaloneRFModel.fit(xTrain,yTrain)

    prediction = abaloneRFModel.predict(xTest)
    mseOos.append(mean_squared_error(yTest, prediction))
(0)

相关推荐