欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > Python代码

TensorFlow保存TensorBoard图像操作

时间:2021-05-18 09:47:32|栏目:Python代码|点击:

简单的代码:

import tensorflow as tf

In [2]: matrix1=tf.constant([[3.,3.]])
In [3]: matrix2=tf.constant([[2.],[2.]])

with tf.Session() as sess:
  ...:   writer = tf.summary.FileWriter('./graph', sess.graph)
  ...:   result = sess.run(tf.matmul(matrix1, matrix2))
  ...:   writer.close()

ipython中使用!+命令可以直接运行terminal命令。

terminal输入: tensorboard --logdir graph/

跳出:Starting TensorBoard 54 at http://amax:6006

在浏览器输入地址加端口号并在graph中查看。

补充知识:tensorflow 利用保存的meta图文件生成log供tensorboard可视化 保存恢复模型

tensorboard可视化图:

import tensorflow as tf
g = tf.Graph() 
with g.as_default() as g: 
  tf.train.import_meta_graph('criteo_80.meta') 
 
with tf.Session(graph=g) as sess: 
  file_writer = tf.summary.FileWriter(logdir='./', graph=g)

保存恢复模型:

# 建模型
saver = tf.train.Saver()
 
with tf.Session() as sess:
  # 存模型,注意此处的model是文件名非路径
  saver.save(sess, "/tmp/model")
 
with tf.Session() as sess:
  # 恢复模型
  saver.restore(sess, "/tmp/model")
# 先恢复图
saver = tf.train.import_meta_graph("/tmp/model.meta")
 
with tf.Session() as sess:
  # 再恢复参数
  saver.restore(sess, "/tmp/model")

上一篇:Python爬虫获取op.gg英雄联盟英雄对位胜率的源码

栏    目:Python代码

下一篇:Python爬虫入门教程01之爬取豆瓣Top电影

本文标题:TensorFlow保存TensorBoard图像操作

本文地址:http://www.codeinn.net/misctech/123944.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有