Spec-Zone.ru › TensorFlow 2.3

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)
Аргументы
x Объект numpy-массива или словарь объектов numpy-массивов. Если массив, массив будет обрабатываться как единственный признак.
y Объект numpy-массива или словарь объекта numpy-массива. None если отсутствует.
batch_size Целое число, размер батчей для возврата.
num_epochs Целое число, количество эпох для обработки данных. Если None будет выполняться бесконечно.
shuffle Булево значение, если True, перемешивает очередь. Избегайте перемешивания во время прогнозирования.
queue_capacity Целое число, размер очереди для накопления.
num_threads Целое число, количество потоков, используемых для чтения и добавления в очередь. Для того чтобы иметь предсказуемый и повторяемый порядок чтения и добавления в очередь, например, в режиме прогнозирования и оценки, num_threads должно быть равно 1.
Возвращаемое значение
Функция, имеющая сигнатуру ()->(словарь features, targets)
Исключения
ValueError если форма y не совпадает с формой значений в x (т.е., значения в x имеют одинаковую форму).
ValueError если дубликаты ключей присутствуют как в x, так и в y, когда y является словарем.
ValueError если x или y являются пустыми словарями.
TypeError x не является словарем или массивом.
ValueError если 'shuffle' не указан или не является булевым значением.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/estimator/inputs/numpy_input_fn

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API