Tensorboard存数据代码解释

简介: Tensorboard存数据代码解释

记录tensorboard存数据代码功能

如果你想要存放一个变量的话,你需要先定义一下,这里我创建了一个定义平均奖励的变量:

self.average_reward = tf.Variable(0.1, trainable=False)

之后我们将这个变量加到tensorboard的summary里面,这样的话,我们之后就需要运行这个节点,然后把值传入进去,tensorboard就会自动帮我们加载。

1. self.average_reward = model.average_reward
2. self.aver_rewards = tf.summary.scalar('Average-Reward', self.average_reward)

我们还需要定义保存的路径:

self.summary_writer = tf.summary.FileWriter("logs/", self.sess.graph)

这样的话我们开始的步骤就完成了,之后我们想要存数据的话,我们可以通过下面的方式去调用就可以了。

1. average_r = np.mean(average_rewards)
2. aver_rewards = self.sess.run(self.aver_rewards, {self.average_reward: average_r})
3. self.summary_writer.add_summary(aver_rewards, self.step)

我们先计算出我们需要tensorboard记录的数据,之后我们run一下我们之前定义的scalar节点,然后把scalar需要的数据传入进去即可,最后我们将数据和训练步数add进summary里面就可以啦。

相关文章
|
19天前
|
数据挖掘 Python
Python示例,展示如何找到最近一次死叉之后尚未形成金叉的位置
【10月更文挑战第7天】金融分析中,“死叉”指短期移动平均线(如MA5)跌破长期移动平均线(如MA10),而“金叉”则相反。本文提供Python代码示例,用于找出最近一次死叉后未形成金叉的位置,涵盖移动平均线计算、交叉点判断及结果输出等步骤,适合金融数据分析。
31 4
|
19天前
|
数据采集 Python
Python实用记录(七):通过retinaface对CASIA-WebFace人脸数据集进行清洗,并把错误图路径放入txt文档
使用RetinaFace模型对CASIA-WebFace人脸数据集进行清洗,并将无法检测到人脸的图片路径记录到txt文档中。
34 1
|
25天前
|
存储 并行计算 PyTorch
探索PyTorch:模型的定义和保存方法
探索PyTorch:模型的定义和保存方法
|
3月前
|
JSON C语言 数据格式
Python导出隐马尔科夫模型参数到JSON文件C语言读取
Python导出隐马尔科夫模型参数到JSON文件C语言读取
26 1
|
12月前
Transformer时间序列预测-多变量输入-单变量输出+多变量输出,完整代码数据,可直接运行
Transformer时间序列预测-多变量输入-单变量输出+多变量输出,完整代码数据,可直接运行
254 0
|
存储 JSON 数据可视化
python之数据的存储,保存,以及生成和重构
python之数据的存储,保存,以及生成和重构
|
PyTorch 算法框架/工具 Python
pytorch保存参数及模型的两种方式
pytorch保存参数及模型的两种方式
576 0
|
机器学习/深度学习 PyTorch 算法框架/工具
PyTorch学习笔记(五):模型定义、修改、保存
Module 类是 torch.nn 模块里提供的一个模型构造类 (nn.Module),是所有神经⽹网络模块的基类,我们可以继承它来定义我们想要的模型; PyTorch模型定义应包括两个主要部分:各个部分的初始化(_init_);数据流向定义(forward) 基于nn.Module,可以通过Sequential,ModuleList和ModuleDict三种方式定义PyTorch模型。
767 0
|
Python
Python批量改变图片名字
Python批量改变图片名字
335 0