tf.compat.v1.estimator.inputs.numpy_input_fn

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

tf.compat.v1.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)