如何关闭fbprophet调用fit方法时输出的迭代训练日志
禁用fbprophet 0.7.1版本拟合迭代日志的方法
你遇到的迭代日志是底层依赖的pystan 2.x在优化参数时的默认输出,针对你使用的版本,可通过以下两种方式禁用:
- 方案1:
fit方法添加verbose参数(最简便)
直接在调用fit方法时传入verbose=0即可关闭迭代输出,修改后的代码如下:
model = Prophet(daily_seasonality = False, weekly_seasonality = False) for regressor in extra_regressors: model.add_regressor(regressor) # verbose设为0关闭所有迭代日志,设为1仅输出关键信息,设为2输出全量日志 model.fit(df_train, verbose=0)
- 方案2:上下文管理器临时屏蔽标准输出(适配特殊场景)
如果加verbose=0后仍有少量残留输出,可通过Python上下文管理器临时接管标准输出,完全屏蔽拟合过程的打印内容:
import contextlib import io model = Prophet(daily_seasonality = False, weekly_seasonality = False) for regressor in extra_regressors: model.add_regressor(regressor) # 临时屏蔽所有stdout输出 with contextlib.redirect_stdout(io.StringIO()): model.fit(df_train)
内容的提问来源于stack exchange,提问作者meliksahturker
相关产品推荐
相关产品推荐

