tf.estimator.inputs.numpy_input_fn

Возвращает функцию ввода, которая будет подавать словарь массивов NumPy в модель.

tf.estimator.inputs.numpy_input_fn(
    x, y=None, batch_size=128, num_epochs=1, shuffle=None, queue_capacity=1000,
    num_threads=1
)

Это возвращает функцию, которая выводит features и targets на основе словаря массивов NumPy. Словарь features имеет те же ключи, что и x. Словарь targets имеет те же ключи, что и y, если y является словарем.

Пример:

age = np.arange(4) * 1.0
height = np.arange(32, 36)
x = {'age': age, 'height': height}
y = np.arange(-32, -28)

with tf.Session() as session:
  input_fn = numpy_io.numpy_input_fn(
      x, y, batch_size=2, shuffle=False, num_epochs=1)