python - 停止显示适合模型的报告

Question

我正在使用 XGGRegressor 构建管道。在逐步拟合模型中，kaggle 显示：

[0] validation_0-rmse:195852.50000
[1] validation_0-rmse:194177.78125
[2] validation_0-rmse:192879.96875
[3] validation_0-rmse:191438.50000
[4] validation_0-rmse:189879.57812
[5] validation_0-rmse:188803.59375
[6] validation_0-rmse:187077.00000
[7] validation_0-rmse:185477.14062
[8] validation_0-rmse:184178.45312
[9] validation_0-rmse:182755.70312
[10]    validation_0-rmse:181168.15625
[11]    validation_0-rmse:180261.75000
[12]    validation_0-rmse:178957.35938
[13]    validation_0-rmse:177349.95312
[14]    validation_0-rmse:176673.50000
...

它是如此之长，有时失去了我的时间。如何将其隐藏在运行代码中？

标签： pythonscikit-learnkaggle

Answer 1

您可以将verbosityof设置xgboost为 0 表示静默，1 表示警告，2 表示信息，3 表示调试，如下所示：

import xgboost as xgb
xgb.set_config(verbosity=1)