XGBoost和pytorch的进度条

2024-03-29 05:50:13 发布

您现在位置:Python中文网/ 问答频道 /正文

有没有一种方法可以为使用xgboost和pytorch的培训模型实现进度条?。 如果没有,有没有办法估计训练时间?在


Tags: 方法进度条模型时间pytorch办法xgboost
2条回答
  1. 如果您使用的是python笔记本,可以尝试以下操作:

    import ipywidgets as widgets
    loading = widgets.IntProgress(value=0, min=0, max=100, step=1, description='Loading:')
    display(loading)
    
    for i in range(100):
        time.sleep(0.05)
        loading.value = i
    

您可以引用tdqm

这样使用:

from tqdm import tqdm
for i in tqdm(range(10000)):
    ...

相关问题 更多 >