在tensorflow get started代码中:
import tensorflow as tf
import numpy as np
features = [tf.contrib.layers.real_valued_column("x", dimension=1)]
estimator = tf.contrib.learn.LinearRegressor(feature_columns=features)
x = np.array([1., 2., 3., 4.])
y = np.array([0., -1., -2., -3.])
input_fn = tf.contrib.learn.io.numpy_input_fn({"x":x}, y, batch_size=4, num_epochs=1000)
estimator.fit(input_fn=input_fn, steps=1000)
estimator.evaluate(input_fn=input_fn)我知道batch_size是什么意思,但是当只有4个训练示例时,num_epochs和steps分别意味着什么?
发布于 2017-04-17 17:28:17
一个时代意味着使用你拥有的全部数据。
步骤意味着使用单个批处理数据。
那么,n_steps = Number of data in single epoch // batch_size.
据docs/python/tf/cont肋骨/学/可培训称,
发布于 2017-12-20 13:35:40
num_epochs指示input_fn将返回整个批处理多少次
steps表示函数应该运行多少次。
对于这里的对象estimator的方法,它将停止运行超过“步骤”次数,或者input_fn停止提供数据,就像Tensorflow API所说的那样。
对于每个步骤,调用input_fn,它返回一批数据。求值直到:-步骤批处理,或- input_fn引发输入结束异常(OutOfRangeError或StopIteration)。
https://stackoverflow.com/questions/43454004
复制相似问题