首页 > 其他 > 详细

Tensorflow 框架的一些细节

时间:2020-07-08 15:38:58      阅读:53      评论:0      收藏:0      [点我收藏+]

其中常用的fetches和feed_dict就是常用的传入参数。fetches主要指从计算图中取回计算结果进行放回的那些placeholder和变量,而feed_dict则是将对应的数据传入计算图中占位符,它是字典数据结构只在调用方法内有效。

DIEN 的代码在 model.train() 和 model.calculate() 中都用到了 session.run() 函数,

  • model.train([self.loss, self.accuracy, self.optimizer], feed_dict={...})
  • model.calculate([self.y_hat, self.loss, self.accuracy], feed_dict={...})

在这里,train() 要进行模型的优化,所以需要输出 self.optimizer, 保证 self.optimizer 这一变量进行了更新
calculate() 实际上是对应测试,需要输出 self.y_hat, 但并不需要进行模型的优化,迭代一个梯度步

Tensorflow 框架的一些细节

原文:https://www.cnblogs.com/Gelthin2017/p/13267001.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!