python神经网络之批量学习tf.train.batch函数示例
更新时间：2022年05月04日 11:00:36 作者：Bubbliiiing
这篇文章主要为大家介绍了python神经网络之批量学习tf.train.batch函数示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
学习前言
当我在快乐的学习SSD训练部分的时候，我发现了一个batch我看不太懂，主要是因为tfrecords的数据读取方式我不理解，所以好好学一下batch吧
tf.train.batch函数
tf.train.batch(
tensors,
batch_size,
num_threads=1,
capacity=32,
enqueue_many=False,
shapes=None,
dynamic_pad=False,
allow_smaller_final_batch=False,
shared_name=None,
name=None
)
其中：
1、tensors：利用slice_input_producer获得的数据组合。
2、batch_size：设置每次从队列中获取出队数据的数量。
3、num_threads：用来控制线程的数量，如果其值不唯一，由于线程执行的特性，数据获取可能变成乱序。
4、capacity：一个整数，用来设置队列中元素的最大数量
5、allow_samller_final_batch：当其为True时，如果队列中的样本数量小于batch_size，出队的数量会以最终遗留下来的样本进行出队；当其为False时，小于batch_size的样本不会做出队处理。
6、name：名字
测试代码
1、allow_samller_final_batch=True
import pandas as pd
import numpy as np
import tensorflow as tf
# 生成数据
def generate_data():
num = 18
label = np.arange(num)
return label
# 获取数据
def get_batch_data():
label = generate_data()
input_queue = tf.train.slice_input_producer([label], shuffle=False,num_epochs=2)
label_batch = tf.train.batch(input_queue, batch_size=5, num_threads=1, capacity=64,allow_smaller_final_batch=True)
return label_batch
# 数据组
label = get_batch_data()
sess = tf.Session()
# 初始化变量
sess.run(tf.global_variables_initializer())
sess.run(tf.local_variables_initializer())
# 初始化batch训练的参数
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess,coord)
try:
while not coord.should_stop():
# 自动获取下一组数据
l = sess.run(label)
print(l)
except tf.errors.OutOfRangeError:
print('Done training')
finally:
coord.request_stop()
coord.join(threads)
sess.close()
运行结果为：
[0 1 2 3 4]
[5 6 7 8 9]
[10 11 12 13 14]
[15 16 17 0 1]
[2 3 4 5 6]
[ 7 8 9 10 11]
[12 13 14 15 16]
[17]
Done training
2、allow_samller_final_batch=False
相比allow_samller_final_batch=True，输出结果少了[17]
import pandas as pd
import numpy as np
import tensorflow as tf
# 生成数据
def generate_data():
num = 18
label = np.arange(num)
return label
# 获取数据
def get_batch_data():
label = generate_data()
input_queue = tf.train.slice_input_producer([label], shuffle=False,num_epochs=2)
label_batch = tf.train.batch(input_queue, batch_size=5, num_threads=1, capacity=64,allow_smaller_final_batch=False)
return label_batch
# 数据组
label = get_batch_data()
sess = tf.Session()
# 初始化变量
sess.run(tf.global_variables_initializer())
sess.run(tf.local_variables_initializer())
# 初始化batch训练的参数
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess,coord)
try:
while not coord.should_stop():
# 自动获取下一组数据
l = sess.run(label)
print(l)
except tf.errors.OutOfRangeError:
print('Done training')
finally:
coord.request_stop()
coord.join(threads)
sess.close()
运行结果为：
[0 1 2 3 4]
[5 6 7 8 9]
[10 11 12 13 14]
[15 16 17 0 1]
[2 3 4 5 6]
[ 7 8 9 10 11]
[12 13 14 15 16]
Done training
以上就是python神经网络之批量学习tf.train.batch函数示例的详细内容，更多关于python神经网络tf.train.batch的资料请关注脚本之家其它相关文章！
相关文章
今天小编就为大家分享一篇Anaconda下安装mysql-python的包实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-06-06
这篇文章主要介绍了关于Python 位运算防坑指南，小编将剧烈向大家说明并且列举python及C#两种语言,需要的朋友可以参考下面文章的具体内容2021-09-09
今天小编就为大家分享一篇pytorch中nn.Conv1d的用法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要为大家详细介绍了python实现矩阵打印的相关代码，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-03-03
本篇文章将涵盖 TensorFlow 的高级应用，包括如何自定义模型的保存和加载过程，以及如何进行分布式训练，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-07-07
这篇文章主要介绍了python 地图经纬度转换、纠偏的实例代码,非常不错，具有一定的参考借鉴价值，需要的朋友可以参考下2018-08-08
这篇文章主要介绍了Python小红书旋转验证码识别实战教程,本文通过示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧2023-08-08
这篇文章主要为大家详细介绍了Windows 7下Python Web环境搭建图文教程，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-03-03
这篇文章主要介绍了Python基于百度AI实现OCR文字识别,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-04-04
自从python3推出关于异步编程的新语法之后，关于异步web框架也是如雨后春笋一般爆发，关于 异步框架的性能也日渐激烈。今天就整理关于 python 的异步框架。2021-06-06
最新评论