tf.contrib.data.read_batch_features
Считывает пакеты Examples. (устарело)
tf.contrib.data.read_batch_features(
file_pattern, batch_size, features, reader=tf.data.TFRecordDataset,
reader_args=None, randomize_input=True, num_epochs=None, capacity=10000
)
Пример:
serialized_examples = [
features {
feature { key: "age" value { int64_list { value: [ 0 ] } } }
feature { key: "gender" value { bytes_list { value: [ "f" ] } } }
feature { key: "kws" value { bytes_list { value: [ "code", "art" ] } } }
},
features {
feature { key: "age" value { int64_list { value: [] } } }
feature { key: "gender" value { bytes_list { value: [ "f" ] } } }
feature { key: "kws" value { bytes_list { value: [ "sports" ] } } }
}
]
Можно использовать следующие аргументы:
features: {
"age": FixedLenFeature([], dtype=tf.int64, default_value=-1),
"gender": FixedLenFeature([], dtype=tf.string),
"kws": VarLenFeature(dtype=tf.string),
}
И ожидаемый результат:
{
"age": [[0], [-1]],
"gender": [["f"], ["f"]],
"kws": SparseTensor(
indices=[[0, 0], [0, 1], [1, 0]],
values=["code", "art", "sports"]
dense_shape=[2, 2]),
}
| Аргументы | |
|---|---|
file_pattern | Список файлов или шаблонов путей к файлам, содержащим Example записи. См. tf.io.gfile.glob для правил шаблонов. |
batch_size | Целое число, представляющее количество записей, объединяемых в одном пакете. |
features | Отображение ключей признаков на FixedLenFeature или VarLenFeature значения. См. tf.io.parse_example. |
reader | Функция или класс, который можно вызвать с filenames тензором и (необязательными) reader_args и который возвращает Dataset из Example тензоров. По умолчанию tf.data.TFRecordDataset. |
reader_args | Дополнительные аргументы для передачи классу читателя. |
randomize_input | Нужно ли рандомизировать вход. |
num_epochs | Целое число, указывающее количество проходов по набору данных. Если None, циклически проходит по набору данных бесконечно. |
capacity | Размер буфера ShuffleDataset. Большая емкость обеспечивает лучшее перемешивание, но увеличивает использование памяти и время запуска. |
| Возвращает | |
|---|---|
Словарь, сопоставляющий ключи в features с Tensor или SparseTensor объектами. |
© 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/r1.15/api_docs/python/tf/contrib/data/read_batch_features