在tensorflow中实现去除不足一个batch的数据


Posted in Python onJanuary 20, 2020

我就废话不多说了,直接上代码吧!

#-*- coding:utf-8 -*-
import tensorflow as tf
import numpy as np
 
value1 = tf.placeholder(dtype=tf.float32)
value2 = tf.placeholder(dtype=tf.float32)
value3 = value1 + value2
 
#定义的dataset有参数,只能使用参数化迭代器
dataset = tf.data.Dataset.range(10)
# 定义参数化迭代器
dataset = dataset.shuffle(100)
dataset = dataset.apply(tf.contrib.data.batch_and_drop_remainder(3)) #每个batch3个数据,不足3个舍弃
iterator = dataset.make_initializable_iterator()
next_element = iterator.get_next()
 
with tf.Session() as sess:
  # 需要用参数初始化迭代器
  for i in range(2):
    sess.run(iterator.initializer)
    while True:
      try:
        value = sess.run(next_element)
        result = sess.run(value3,feed_dict={value1:value,value2:value})
        print(result)
      except tf.errors.OutOfRangeError:
        print("End of epoch %d" % i)
        break

以上这篇在tensorflow中实现去除不足一个batch的数据就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python AES加密模块用法分析
May 22 Python
python 接收处理外带的参数方法
Dec 03 Python
python调用动态链接库的基本过程详解
Jun 19 Python
对numpy下的轴交换transpose和swapaxes的示例解读
Jun 26 Python
python-序列解包(对可迭代元素的快速取值方法)
Aug 24 Python
Python中Unittest框架的具体使用
Aug 27 Python
python RC4加密操作示例【测试可用】
Sep 26 Python
Python实现括号匹配方法详解
Feb 10 Python
python操作yaml说明
Apr 08 Python
python 实现简单的计算器(gui界面)
Nov 11 Python
Python基础详解之描述符
Apr 28 Python
Python socket如何解析HTTP请求内容
Feb 12 Python
Tensorflow实现在训练好的模型上进行测试
Jan 20 #Python
Python线程条件变量Condition原理解析
Jan 20 #Python
tensorflow tf.train.batch之数据批量读取方式
Jan 20 #Python
Python list运算操作代码实例解析
Jan 20 #Python
Python模块future用法原理详解
Jan 20 #Python
使用Tensorflow将自己的数据分割成batch训练实例
Jan 20 #Python
Python JSON编解码方式原理详解
Jan 20 #Python
You might like
PHP array_multisort() 函数的深入解析
2013/06/20 PHP
PHP扩展模块memcached长连接使用方法分析
2014/12/24 PHP
php微信开发之上传临时素材
2016/06/24 PHP
PHP编程计算两个时间段是否有交集的实现方法(不算边界重叠)
2017/05/30 PHP
php如何修改SESSION的生存存储时间的实例代码
2017/07/05 PHP
PHP PDOStatement::rowCount讲解
2019/02/01 PHP
JS事件Event元素(兼容IE,Firefox,Chorme)
2012/11/01 Javascript
js实现鼠标感应向下滑动隐藏菜单的方法
2015/02/20 Javascript
【经典源码收藏】基于jQuery的项目常见函数封装集合
2016/06/07 Javascript
angularjs的select使用及默认选中设置
2017/04/08 Javascript
Vue中使用clipboard实现复制功能
2018/09/05 Javascript
react+ant design实现Table的增、删、改的示例代码
2018/12/27 Javascript
js的各种数据类型判断的介绍
2019/01/19 Javascript
JavaScript基于遍历操作实现对象深拷贝功能示例
2019/03/05 Javascript
Node.js + express实现上传大文件的方法分析【图片、文本文件】
2019/03/14 Javascript
Python之py2exe打包工具详解
2017/06/14 Python
在Python 中实现图片加框和加字的方法
2019/01/26 Python
基于h5py的使用及数据封装代码
2019/12/26 Python
python对指定字符串逆序的6种方法(小结)
2020/04/02 Python
哈工大自然语言处理工具箱之ltp在windows10下的安装使用教程
2020/05/07 Python
Tensorflow--取tensorf指定列的操作方式
2020/06/30 Python
解决TensorFlow训练模型及保存数量限制的问题
2021/03/03 Python
Groupon荷兰官方网站:高达70%的折扣
2019/11/01 全球购物
毕业生幼师求职自荐信
2013/10/01 职场文书
工作睡觉检讨书
2014/02/25 职场文书
2014乡镇“三八”国际劳动妇女节活动总结
2014/03/01 职场文书
松材线虫病防治方案
2014/06/15 职场文书
股东授权委托书范本
2014/09/13 职场文书
乡镇党员群众路线教育实践活动对照检查材料思想汇报
2014/10/05 职场文书
北京天坛导游词
2015/02/12 职场文书
幼儿园综治宣传月活动总结
2015/05/07 职场文书
优秀团员主要事迹范文
2015/11/05 职场文书
css3 利用transform-origin 实现圆点分布在大圆上布局及旋转特效
2021/04/29 HTML / CSS
Html5大屏数据可视化开发的实现
2021/06/11 HTML / CSS
Nginx限流和黑名单配置
2022/05/20 Servers
CentOS7设置ssh服务以及端口修改方式
2022/12/24 Servers