Keras使用tensorboard显示训练过程的实例


Posted in Python onFebruary 15, 2020

众所周知tensorflow造势虽大却很难用,因此推荐使用Keras,它缺省是基于tensorflow的,但通过修改keras.json也可以用于theano。但是为了能用tensorflow提供的tensorboard,因此建议仍基于tensorflow。

那么问题来了,由于Keras隐藏了tensorflow那令人诟病、可笑至极的graph构建方法,那么如何使用tensorboard呢?一般网站上会告诉你是这样的:

方法一(标准调用方法):

采用keras特有的fit()进行训练,只要在fit的时候指定callbacks函数即可,代码如下

from keras.callbacks import TensorBoard 
from keras.models import Sequential 
…… 
model = Sequential() 
…… 
tbCallBack = keras.callbacks.TensorBoard(log_dir='./Graph', 
histogram_freq=1, 
write_graph=True, 
write_images=True)

model_history = model.fit(X_train_train, 
y_train_train, 
batch_size=batch_size, 
epochs=epochs, 
verbose=1, 
validation_data=(X_train_val, y_train_val), 
callbacks = [EarlyStopping(patience=patience,mode='min',verbose=1), 
history, 
tbCallBack])

虽然这种方法看上去很美,但它只适用于标准训练方法,如果你想用自己的训练方法,就需要调用train_on_batch,而不能直接使用fit(),这时就要采用下面这种方法:

方法二(特殊调用方法):

这种方法可用于调用train_on_batch的情况。

%预先写好writer,定义好model 
writer = tf.summary.FileWriter(…) 
model = … 
%训练时 
loss = model.train_on_batch(…) 
summary = tf.Summary(value=[ 
tf.Summary.Value(tag=”d_loss”, simple_value=d_loss), 
tf.Summary.Value(tag=”g_loss”, simple_value=g_loss), 
]) 
writer.add_summary(summary)

虽然很简单,但这种方法只能显示scalar类型,不能显示image,histgram等,非常不实用。真正实用的是下面的终结方法:

方法三(最实用的方法)

最实用的还是用tensorflow原生的调用方法,虽然相对方法二麻烦一点,但考虑到此方法与tensorflow一样,不需要去记那些额外的花拳绣腿,因此反而是最简单的,也是最有效的。

代码如下:

import tensorflow as tf 
import datetime 
%在训练开始之前,预先定义好可视化的东西,用的是原生的tensorflow方法,这里我们以一个GAN模型为例,让它显示整张模型图,两个标量损失函数,以及5个生成图像。方法是预先用placeholder声明所要显示的那些东西,然后在训练过程中将训练结果来填充它们。你可以自行加入histgram,都和tensorflow一样 
#start tensorboard 
sess=tf.Session() 
logdir = “tensorboard/” + datetime.datetime.now().strftime(“%Y%m%d-%H%M%S”) + “/” 
writer = tf.summary.FileWriter(logdir, sess.graph) 
D_LOSS = tf.placeholder(tf.float32, []) 
G_LOSS = tf.placeholder(tf.float32, []) 
IMAGES = tf.placeholder(tf.float32,shape=[None,28,28,1]) 
tf.summary.scalar(“D_LOSS”, D_LOSS) 
tf.summary.scalar(“G_LOSS”, G_LOSS) 
tf.summary.image(“IMAGES”, IMAGES, 5) 
merged=tf.summary.merge_all() 
#end tensorboard

训练迭代过程中,是这样的

for epoch in range(100): 
% 用keras的train_on_batch方法进行训练 
d_loss = d.train_on_batch(。。。。。。) 
g_loss = d_on_g.train_on_batch(。。。。。。) 
generated_images = g.predict(。。。。。。) 
if index%10==0: #tensorboard 
% 将训练结果填充可视化数据 
summary=sess.run(merged,feed_dict={D_LOSS:d_loss, G_LOSS:g_loss, IMAGES:generated_images}) 
writer.add_summary(summary,index)

以上这篇Keras使用tensorboard显示训练过程的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中with及contextlib的用法详解
Jun 08 Python
在PyCharm环境中使用Jupyter Notebook的两种方法总结
May 24 Python
pandas使用get_dummies进行one-hot编码的方法
Jul 10 Python
python多线程并发实例及其优化
Jun 27 Python
react+django清除浏览器缓存的几种方法小结
Jul 17 Python
python解析多层json操作示例
Dec 30 Python
python 爬取马蜂窝景点翻页文字评论的实现
Jan 20 Python
django 解决model中类写不到数据库中,数据库无此字段的问题
May 20 Python
Tensorflow卷积实现原理+手写python代码实现卷积教程
May 22 Python
python 在sql语句中使用%s,%d,%f说明
Jun 06 Python
Python3爬虫关于代理池的维护详解
Jul 30 Python
python爬虫搭配起Bilibili唧唧的流程分析
Dec 01 Python
Python利用逻辑回归分类实现模板
Feb 15 #Python
python发qq消息轰炸虐狗好友思路详解(完整代码)
Feb 15 #Python
利用Tensorboard绘制网络识别准确率和loss曲线实例
Feb 15 #Python
tensorboard显示空白的解决
Feb 15 #Python
解决Tensorboard可视化错误:不显示数据 No scalar data was found
Feb 15 #Python
tensorboard 可以显示graph,却不能显示scalar的解决方式
Feb 15 #Python
解决Tensorboard 不显示计算图graph的问题
Feb 15 #Python
You might like
LotusPhp笔记之:基于ObjectUtil组件的使用分析
2013/05/06 PHP
php倒计时出现-0情况的解决方法
2016/07/28 PHP
PHP FileSystem 文件系统常用api整理总结
2019/07/12 PHP
document.write的几点使用心得
2014/05/14 Javascript
jQuery表格行上移下移和置顶的实现方法
2015/10/08 Javascript
javascript如何实现360度全景照片问题汇总
2016/04/04 Javascript
Angular4学习笔记之新建项目的方法
2017/07/18 Javascript
JS沙箱模式实例分析
2017/09/04 Javascript
Vue 仿QQ左滑删除组件功能
2018/03/12 Javascript
vue+axios+mock.js环境搭建的方法步骤
2018/08/28 Javascript
解决vue移动端适配问题
2018/12/12 Javascript
Vue组件间通信方法总结(父子组件、兄弟组件及祖先后代组件间)
2019/04/17 Javascript
layui实现数据表格table分页功能(ajax异步)
2019/07/27 Javascript
Vue formData实现图片上传
2019/08/20 Javascript
element-ui 远程搜索组件el-select在项目中组件化的实现代码
2019/12/04 Javascript
React 父子组件通信的实现方法
2019/12/05 Javascript
Python2.x版本中基本的中文编码问题解决
2015/10/12 Python
详解python如何调用C/C++底层库与互相传值
2016/08/10 Python
利用python获取Ping结果示例代码
2017/07/06 Python
Python中max函数用于二维列表的实例
2018/04/03 Python
如何用Python实现简单的Markdown转换器
2018/07/16 Python
浅谈python之新式类
2018/08/12 Python
python pygame实现五子棋小游戏
2020/10/26 Python
Python qqbot 实现qq机器人的示例代码
2019/07/11 Python
解决pycharm启动后总是不停的updating indices...indexing的问题
2019/11/27 Python
Pytorch实现LSTM和GRU示例
2020/01/14 Python
python中的selenium安装的步骤(浏览器自动化测试框架)
2020/03/17 Python
文件上传服务器-jupyter 中python解压及压缩方式
2020/04/22 Python
旷课检讨书3000字
2014/02/04 职场文书
美丽乡村建设实施方案
2014/03/23 职场文书
文秘求职信范文
2014/04/10 职场文书
学雷锋演讲稿汇总
2014/05/10 职场文书
2014大学班主任工作总结
2014/11/08 职场文书
云冈石窟导游词
2015/02/04 职场文书
紫日观后感
2015/06/05 职场文书
SQL Server #{}可以防止SQL注入
2022/05/11 SQL Server