如何禁用NeuralProphet训练时默认显示的tqdm进度条
禁用NeuralProphet训练默认进度条
需求
训练NeuralProphet算法时,需要隐藏界面默认出现的绿色进度条,待隐藏的效果如下图:
测试示例代码
import pandas as pd from neuralprophet import NeuralProphet, set_log_level set_log_level("ERROR") data_location = "https://raw.githubusercontent.com/ourownstory/neuralprophet-data/main/datasets/" df = pd.read_csv(data_location + "air_passengers.csv") #df.tail(3) m = NeuralProphet(n_lags=5, n_forecasts=3) metrics_train = m.fit(df=df, freq="MS",progress='print')
已尝试操作
已将日志级别设置为ERROR,仍无法关闭进度条,需要彻底禁用该默认显示的进度条。
解决方法
进度条不受日志级别参数控制,由fit()方法的progress参数直接管控,当前代码里传入progress='print'是主动开启进度输出的配置,只需要将该参数修改为progress=None即可彻底关闭所有训练进度条,修改后的fit调用代码如下:
metrics_train = m.fit(df=df, freq="MS", progress=None)
progress参数的可选配置说明:
None:完全禁用所有进度相关输出"print":输出逐行训练记录+绿色进度条"plot":仅在Jupyter等交互环境生效,展示动态训练曲线"auto":默认值,自动识别运行环境选择匹配的进度展示形式
内容的提问来源于stack exchange,提问作者sabarilals
相关产品推荐
相关产品推荐

