汇总记录一下在 Jupyter Notebook 中使用过的两个实时绘制损失和准确率的工具,均来自 微博爱可可-爱生活 的推荐。
livelossplot
这款工具用于实时绘制训练时的损失和准确率,方便好用,不需要自己另外再写 plot 函数。Keras
和 PyTorch
中都可以使用。之前推荐过给朋友,最近自己才用上,感觉真的超方便了~
from livelossplot import PlotLossesKeras
model.fit(X_train, Y_train,
epochs=10,
validation_data=(X_test, Y_test),
callbacks=[PlotLossesKeras()],
verbose=0)
fastprogress
这款工具由 Fast.ai 出品。同时有进度条显示和绘制损失,准确率的功能。
from fastprogress import master_bar, progress_bar
from time import sleep
mb = master_bar(range(10))
for i in mb:
for j in progress_bar(range(100), parent=mb):
sleep(0.01)
mb.child.comment = f'second bar stat'
mb.first_bar.comment = f'first bar stat'
mb.write(f'Finished loop {i}.')
#mb.update_graph(graphs, x_bounds, y_bounds)
网友评论