本文转自:https://www.jianshu.com/p/a9247add0046

livelossplot

这款工具用于实时绘制训练时的损失和准确率,方便好用,不需要自己另外再写 plot 函数。Keras 和 PyTorch 中都可以使用。之前推荐过给朋友,最近自己才用上,感觉真的超方便了~ 如下图所示:

实时绘制训练过程中损失和准确率的变化趋势 python keras jupyter notebook

 

用法:

1 from livelossplot import PlotLossesKeras
2 
3 model.fit(X_train, Y_train,
4           epochs=10,
5           validation_data=(X_test, Y_test),
6           callbacks=[PlotLossesKeras()],
7           verbose=0)

fastprogress

这款工具由 Fast.ai 出品。同时有进度条显示和绘制损失,准确率的功能。

 1 from fastprogress import master_bar, progress_bar
 2 from time import sleep
 3 mb = master_bar(range(10))
 4 for i in mb:
 5     for j in progress_bar(range(100), parent=mb):
 6         sleep(0.01)
 7         mb.child.comment = f'second bar stat'
 8     mb.first_bar.comment = f'first bar stat'
 9     mb.write(f'Finished loop {i}.')
10     #mb.update_graph(graphs, x_bounds, y_bounds)

显示效果:

实时绘制训练过程中损失和准确率的变化趋势 python keras jupyter notebook

 

 

 

安装:

  两者使用的时候都要通过命令行:pip install (工具名)