forest_reg = RandomForestRegressor(max_depth=(15),min_samples_leaf=2,min_samples_split=3#,n_estimators=100) param_grid = {'n_estimators': [ 20]} grid_search = GridSearchCV(forest_reg, param_grid, cv=5, scoring='r2') grid_search.fit(X_train_scaled, y_train) best_forest_reg = grid_search.best_estimator_ y_forest_pred_train = best_forest_reg.predict(X_train_scaled) y_forest_pred_test = best_forest_reg.predict(X_test_scaled) print("随机森林模型 R2 (训练集):", r2_score(y_train, y_forest_pred_train)) print("随机森林模型 R2 (测试集):", r2_score(y_test, y_forest_pred_test))。
时间: 2023-11-24 21:07:05 浏览: 29
RandomForest随机森林算法
这是一个使用随机森林回归模型对数据进行训练和测试的代码。其中,使用了网格搜索(GridSearchCV)来寻找最佳的超参数(n_estimators),同时使用了交叉验证(cv=5)来评估模型的性能。最终输出了模型在训练集和测试集上的R2分数。R2分数是评估回归模型性能的一种常用指标,表示模型解释数据方差的能力,取值范围为0到1,值越接近1表示模型性能越好。
阅读全文