在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获取本地计算机名字的方法
Apr 29 Python
python扫描proxy并获取可用代理ip的实例
Aug 07 Python
python selenium UI自动化解决验证码的4种方法
Jan 05 Python
Python+PIL实现支付宝AR红包
Feb 09 Python
Python并行分布式框架Celery详解
Oct 15 Python
Python自动发送邮件的方法实例总结
Dec 08 Python
Python 多线程不加锁分块读取文件的方法
Dec 11 Python
Python 一键制作微信好友图片墙的方法
May 16 Python
python实现图片插入文字
Nov 26 Python
浅谈keras中Dropout在预测过程中是否仍要起作用
Jul 09 Python
如何用用Python将地址标记在地图上
Feb 07 Python
python利用后缀表达式实现计算器功能
Feb 22 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 var_dump遍历对象属性的函数与应用代码
2010/06/04 PHP
PHP strncasecmp字符串比较的小技巧
2011/01/04 PHP
php+mysql大量用户登录解决方案分析
2014/12/29 PHP
PHP结合jQuery.autocomplete插件实现输入自动完成提示的功能
2015/04/27 PHP
PHP验证信用卡卡号是否正确函数
2015/05/27 PHP
推荐20家国外的脚本下载网站
2011/04/28 Javascript
jquery maxlength使用说明
2011/09/09 Javascript
JavaScript单元测试ABC
2012/04/12 Javascript
JavaScript删除数组元素的方法
2015/03/20 Javascript
JS+HTML5实现的前端购物车功能插件实例【附demo源码下载】
2016/10/17 Javascript
Angular2安装angular-cli
2017/05/21 Javascript
浅谈react-native热更新react-native-pushy集成遇到的问题
2017/09/30 Javascript
了解javascript中变量及函数的提升
2019/05/27 Javascript
原生js代码能实现call和bind吗
2019/07/31 Javascript
countup.js实现数字动态叠加效果
2019/10/17 Javascript
LayUi数据表格自定义赋值方式
2019/10/26 Javascript
[02:57]DOTA2亚洲邀请赛小组赛第四日 赛事回顾
2015/02/02 DOTA
python使用tkinter实现简单计算器
2018/01/30 Python
Python3.6笔记之将程序运行结果输出到文件的方法
2018/04/22 Python
python读取和保存图片5种方法对比
2018/09/12 Python
Python/ArcPy遍历指定目录中的MDB文件方法
2018/10/27 Python
不到20行代码用Python做一个智能聊天机器人
2019/04/19 Python
Python使用lambda表达式对字典排序操作示例
2019/07/25 Python
Pytorch之finetune使用详解
2020/01/18 Python
使用keras实现BiLSTM+CNN+CRF文字标记NER
2020/06/29 Python
HTML5 Canvas实现玫瑰曲线和心形图案的代码实例
2014/04/10 HTML / CSS
HTML5 用动画的表现形式装载图像
2016/03/08 HTML / CSS
受欢迎的大学生自我评价
2013/12/05 职场文书
班组长的岗位职责
2013/12/09 职场文书
小学生新学期寄语
2014/01/19 职场文书
公司端午节活动方案
2014/02/04 职场文书
父亲的菜园教学反思
2014/02/13 职场文书
煤矿安全生产责任书
2014/04/15 职场文书
办公经费申请报告
2015/05/15 职场文书
redis 限制内存使用大小的实现
2021/05/08 Redis
浅谈Redis 中的过期删除策略和内存淘汰机制
2022/04/03 Redis