tf.data.experimental.table_from_dataset
Возвращает таблицу поиска, созданную на основе указанного набора данных.
tf.data.experimental.table_from_dataset(
dataset=None,
num_oov_buckets=0,
vocab_size=None,
default_value=None,
hasher_spec=lookup_ops.FastHashSpec,
key_dtype=tf.dtypes.string,
name=None
)
Эта операция создает таблицу поиска на основе указанного набора данных пар (ключ, значение).
Если num_oov_buckets больше нуля, при поиске токена, отсутствующего в словаре, возвращается идентификатор корзины, вычисленный на основе его хеша. В противном случае ему присваивается default_value. Диапазон идентификаторов корзин: [vocabulary size, vocabulary size + num_oov_buckets - 1].
Примеры использования:
keys = tf.data.Dataset.range(100)
values = tf.data.Dataset.range(100).map(
lambda x: tf.strings.as_string(x * 2))
ds = tf.data.Dataset.zip((keys, values))
table = tf.data.experimental.table_from_dataset(
ds, default_value='n/a', key_dtype=tf.int64)
table.lookup(tf.constant([0, 1, 2], dtype=tf.int64)).numpy()
array([b'0', b'2', b'4'], dtype=object)| Аргументы | |
|---|---|
dataset | Набор данных, содержащий пары (ключ, значение). |
num_oov_buckets | Количество корзин для значений, отсутствующих в словаре. |
vocab_size | Количество элементов в словаре, если оно известно. |
default_value | Значение, используемое для элементов признаков, отсутствующих в словаре. По умолчанию — -1. |
hasher_spec | HasherSpec для указания хеш-функции, используемой для распределения значений по корзинам для элементов, отсутствующих в словаре. |
key_dtype | Тип данных key. |
name | Имя этой операции (необязательно). |
| Возвращает | |
|---|---|
| Таблицу поиска, созданную на основе указанного набора данных. |
| Вызывает исключения | |
|---|---|
ValueError | Если
|
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/data/experimental/table_from_dataset