在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并发编程之多进程、多线程、异步和协程详解
Oct 28 Python
Python使用内置json模块解析json格式数据的方法
Jul 20 Python
Python字符串和字典相关操作的实例详解
Sep 23 Python
python实现停车管理系统
Nov 30 Python
python文字和unicode/ascll相互转换函数及简单加密解密实现代码
Aug 12 Python
python3实现从kafka获取数据,并解析为json格式,写入到mysql中
Dec 23 Python
pyMySQL SQL语句传参问题,单个参数或多个参数说明
Jun 06 Python
Python机器学习之基础概述
May 19 Python
Python 线程池模块之多线程操作代码
May 20 Python
python生成可执行exe控制Microsip自动填写号码并拨打功能
Jun 21 Python
如何利用Python实现n*n螺旋矩阵
Jan 18 Python
Python实现视频自动打码的示例代码
Apr 08 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
thinkphp autoload 命名空间自定义 namespace
2015/07/17 PHP
php外部执行命令函数用法小结
2016/10/11 PHP
php实现QQ小程序发送模板消息功能
2019/09/18 PHP
php模拟post提交请求调用接口示例解析
2020/08/07 PHP
基于jquery的大众点评,分类导航实现代码
2011/08/23 Javascript
Jquery为单选框checkbox绑定单击click事件
2012/12/18 Javascript
二叉树先序遍历的非递归算法具体实现
2014/01/09 Javascript
JavaScript的strict模式与with关键字介绍
2014/02/08 Javascript
EasyUI,点击开启编辑框,并且编辑框获得焦点的方法
2015/03/01 Javascript
jQuery实现渐变下拉菜单的简单方法
2015/03/11 Javascript
基于Css3和JQuery实现打字机效果
2015/08/11 Javascript
AngularJS表格详解及示例代码
2016/08/17 Javascript
Vue实现virtual-dom的原理简析
2017/07/10 Javascript
React实践之Tree组件的使用方法
2017/09/30 Javascript
NodeJS简单实现WebSocket功能示例
2018/02/10 NodeJs
开发用到的js封装方法(20种)
2018/10/12 Javascript
vue-quill-editor+plupload富文本编辑器实例详解
2018/10/19 Javascript
vue中img src 动态加载本地json的图片路径写法
2019/04/25 Javascript
微信小程序 SOTER 生物认证DEMO 指纹识别功能
2019/12/13 Javascript
[01:04:49]KG vs LGD 2019国际邀请赛小组赛 BO2 第二场 8.15
2019/08/16 DOTA
python基础教程之字典操作详解
2014/03/25 Python
python中快速进行多个字符替换的方法小结
2016/12/15 Python
python 类详解及简单实例
2017/03/24 Python
python+pandas+时间、日期以及时间序列处理方法
2018/07/10 Python
Python对切片命名的实现方法
2018/10/16 Python
python Web flask 视图内容和模板实现代码
2019/08/23 Python
Python3.6实现根据电影名称(支持电视剧名称),获取下载链接的方法
2019/08/26 Python
Django框架模板用法入门教程
2019/11/04 Python
Python爬虫过程解析之多线程获取小米应用商店数据
2020/11/14 Python
Anaconda的安装与虚拟环境建立
2020/11/18 Python
eBay奥地利站:eBay.at
2019/07/24 全球购物
如何利用find命令查找文件
2015/02/07 面试题
怎样拟定创业计划书
2014/05/01 职场文书
机关党总支领导班子整改方案
2014/09/20 职场文书
总结python多进程multiprocessing的相关知识
2021/06/29 Python
HTML常用标签超详细整理
2022/03/19 HTML / CSS