研究文章
应用梯度提高机器学习算法来预测单轴抗压强度的软在塔尔煤田沉积岩
表4
执行一个GBR框架在python中使用梯度提高回归量。
|
| 描述 |
Python脚本 |
|
| 导入python包 |
从sklearn。整体从sklearn进口GradientBoostingRegressor。model_selection进口cross_val_score RepeatedKFold |
| 解释框架的属性 |
框架= GradientBoostingRegressor () |
| #的解释评价机制 |
| 简历= RepeatedKFold (n_splits = 10, n_repeats = 3, random_state = 1) |
| #框架评估 |
| n_scores = cross_val_score(框架、x_train UCS_train得分=“neg_mean_absolute_error”, |
| 简历=简历,n_jobs = 1) |
| #性能报告 |
| 打印(“美:%。8 f (% .8f) %(意思是(n_scores)、性病(n_scores))) |
| #适合训练数据集上的框架 |
| 框架。fit (x, UCS) |
| 训练数据集 |
Train_dataset_Forecasting = Framework.predict (x_train) |
| 测试数据集 |
Test_dataset_Forecasting = Framework.predict (x_test) |
|
|
