tensorflow
文章平均质量分 60
我爱写报告
这个作者很懒,什么都没留下…
展开
-
tfrecord相关
给自己留档,因为永远记不住用法。 python读tfrecord文件 python读取tfrecord文件有很多种姿势,但大体过程类似: # 定义feature字典,字典key必须在tfrecord中存在,反之则不要求 def parse(example): features = { 'id': tf.FixedLenFeature(1, dtype=tf.string), 'label': tf.FixLenFeature(1, dtype=tf.float32), 'fea原创 2021-06-15 23:17:42 · 175 阅读 · 0 评论 -
tensorflow AUC & streaming_auc
对AUC和它计算方法的解释可以看这位大佬的文章:https://www.cnblogs.com/peizhe123/p/5081559.html,本文主要讲tensorflow中计算AUC的API实现。 tf.contrib.metrics.streaming_auc & tf.metrics.auc 在tf1.x中有两个计算AUC的API: tf.contrib.metrics.streaming_auc和tf.metrics.auc,在使用tf.contrib.metrics.streaming原创 2021-04-07 20:20:50 · 1348 阅读 · 0 评论 -
tf.train.Saver
Saver类用于保存和重载变量到checkpoint。 Checkpoint是一系列有特定格式的二进制文件,这些文件会把变量名和tensor值映射起来。 Saver能够通过一个给定的计数器自动给checkpoint文件排序,这使得我们能够在训练一个模型时,在不同的step下保存多个checkpoint。同时Saver还能够自动管理checkpoint,例如我们可以通过Saver来只保存最近N个ch...翻译 2020-04-30 19:40:43 · 320 阅读 · 0 评论 -
Tensorflow模型保存和重载
最近因为项目要求,需要把模型的训练和测试过程分开,这里主要涉及两个过程:训练图的存取和参数的存取。 以下所有/home/yy/xiajbxie/model是我的模型的存储路径,将其换成你自己的即可。 tf.train.Saver() Saver的作用中文社区已经讲得相当清楚。tf.train.Saver()类的基本操作时save()和restore()函数,分别负责模型参数的保存和恢复。...原创 2018-02-20 17:29:01 · 1379 阅读 · 0 评论 -
yadlt代码笔记:tensorflow实现DBN
在找DBN的tensorflow实现的时候发现了这么一个工具,github主页指路。里面有RNN, CNN, 基于玻尔兹曼机的网络等等模型的实现。官方文档也有不过版本较旧,没太大参考价值了。 安装和使用 可以通过pip安装,但还是建议从github主页下载,毕竟后续无论什么工作都可能涉及到修改源码嘛。下载后进入Deep-Learning-TensorFlow目录,运行安装脚本。 su...原创 2018-01-03 16:45:44 · 11296 阅读 · 22 评论 -
深度学习模型的优化算法及tensorflow实现
模型的优化对于传统机器学习和深度学习都是很重要的,尤其是深度学习,在训练过中很可能需要解决一些更加困难的挑战。目前来说,流行且使用较广泛的优化算法有随机梯度下降,具动量的随机梯度下降,RMSProp算法,具动量的RMSProp,AdaDelta和Adam等,以下会选择几种进行具体说明,大部分抄自《深度学习》和tensorflow官方文档。如有错漏欢迎指正。 随机梯度下降 随机梯度下降(SGD)原创 2018-01-30 21:53:33 · 6372 阅读 · 0 评论 -
Linux下CUDA+CUDNN+TensorFlow安装笔记
之前用过TensorFlow的CPU版,现在买了个新电脑,就想把GPU也用起来,又因为目前大部分工作是在linux下做,所以有了在linux下安装cuda和TensorFlow的计划,快一个礼拜一直在搞这件事,重装了N次系统,cuda和TensorFlow终于是能用了。姑且写一下攻略吧,也算自己留个档。注意事项 此攻略适用于广大使用Nvidia显卡的Ubuntu用户,CentOS,RedHat,Wi原创 2017-12-12 20:22:29 · 16703 阅读 · 2 评论