Jupyter笔记本中带有TQM进度条的Keras模型
keras-tqdm的Python项目详细描述
Keras与TQM进度条的集成。
主要功能
- TQM支持嵌套进度条。如果你有合适的 预测外部TQM循环中的循环,嵌套循环将 显示正确。
- TQM支持Jupyter/iPython笔记本电脑。
- TQM看起来很棒!
TQDMNotebookCallback与leave_inner=False(默认值)
TQDMNotebookCallback与leave_inner=True
TQDMCallback用于命令行脚本
安装
稳定释放
pip install keras-tqdm
开发版本
pip install git+https://github.com/bstriner/keras-tqdm.git --upgrade --no-deps
开发模式(对源代码的更改在不重新安装的情况下生效)
git clone https://github.com/bstriner/keras-tqdm.git cd keras-tqdm python setup.py develop
基本用法
使用Keras TQM非常容易。唯一需要做的更改是删除默认消息(verbose=0)并将回调添加到model.fit。剩下的就自动发生了!对于jupyter笔记本,所需的代码修改非常简单:
from keras_tqdm import TQDMNotebookCallback # keras, model definition... model.fit(X_train, Y_train, verbose=0, callbacks=[TQDMNotebookCallback()])
对于纯文本模式(例如,对于python,从命令行运行)
from keras_tqdm import TQDMCallback # keras, model definition... model.fit(X_train, Y_train, verbose=0, callbacks=[TQDMCallback()])
高级用法
使用keras_tqdm将TQM进度条用于Keras Fit循环。 keras_tqdm循环可以嵌套在tqdm循环中以显示嵌套的进度条(尽管您可以使用它们 在普通的循环中)。 设置verbose=0以取消默认进度条。
from keras_tqdm import TQDMCallback from tqdm import tqdm for model in tqdm(models, desc="Training several models"): model.fit(x, y, verbose=0, callbacks=[TQDMCallback()])
对于ipython和jupyter笔记本TQDMNotebookCallback,而不是TQDMCallback。在自己的代码中使用tqdm_notebook,而不是tqdm。 格式化由python格式字符串控制。默认的metric_format是"{name}: {value:0.3f}"。 例如,使用TQDMCallback(metric_format="{name}:{value:0.6f}")表示6个小数点,或使用{name}: {value:e}表示科学记数法。
有问题吗?
请随时提交减贫战略和问题。评论、问题和 欢迎提出要求。如果你需要更多的控制,子类 TQDMCallback并重写tqdm函数。