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