tf.compat.v1.train.VocabInfo
Информация о словаре для теплого старта.
tf.compat.v1.train.VocabInfo(
new_vocab,
new_vocab_size,
num_oov_buckets,
old_vocab,
old_vocab_size=-1,
backup_initializer=None,
axis=0
)
См. tf.estimator.WarmStartSettings для примеров использования VocabInfo для теплого старта.
Аргументы: new_vocab: [Обязательно] Путь к новому файлу словаря (используется с моделью, которую нужно обучить). new_vocab_size: [Обязательно] Целое число, указывающее, сколько записей нового словаря будет использовано при обучении. num_oov_buckets: [Обязательно] Целое число, указывающее, сколько ведер OOV связаны со словарем. old_vocab: [Обязательно] Путь к старому файлу словаря (используется с контрольной точкой, с которой нужно начать теплый запуск). old_vocab_size: [Необязательно] Целое число, указывающее, сколько записей старого словаря использовалось при создании контрольной точки. Если не указано, будет использован весь старый словарь. backup_initializer: [Необязательно] Инициализатор переменных, используемый для переменных, соответствующих новым записям словаря и OOV. Если не указано, эти записи будут инициализированы нулями. axis: [Необязательно] Указывает, к какому осям соответствует словарь. По умолчанию 0 соответствует наиболее частому случаю использования (вложения или линейные веса для бинарной классификации/регрессии). Ось 1 может быть использована для теплого старта выходных слоев со словарями классов.
Возвращает: VocabInfo, представляющий информацию о словаре для теплого старта.
Возможные исключения: ValueError: axis не равно ни 0, ни 1.
Example Usage:
embeddings_vocab_info = tf.VocabInfo(
new_vocab='embeddings_vocab',
new_vocab_size=100,
num_oov_buckets=1,
old_vocab='pretrained_embeddings_vocab',
old_vocab_size=10000,
backup_initializer=tf.compat.v1.truncated_normal_initializer(
mean=0.0, stddev=(1 / math.sqrt(embedding_dim))),
axis=0)
softmax_output_layer_kernel_vocab_info = tf.VocabInfo(
new_vocab='class_vocab',
new_vocab_size=5,
num_oov_buckets=0, # No OOV for classes.
old_vocab='old_class_vocab',
old_vocab_size=8,
backup_initializer=tf.compat.v1.glorot_uniform_initializer(),
axis=1)
softmax_output_layer_bias_vocab_info = tf.VocabInfo(
new_vocab='class_vocab',
new_vocab_size=5,
num_oov_buckets=0, # No OOV for classes.
old_vocab='old_class_vocab',
old_vocab_size=8,
backup_initializer=tf.compat.v1.zeros_initializer(),
axis=0)
#Currently, only axis=0 and axis=1 are supported.
```
<!-- Tabular view -->
<table class="responsive fixed orange">
<colgroup><col width="214px"><col></colgroup>
<tr><th colspan="2"><h2 class="add-link">Attributes</h2></th></tr>
<tr>
<td>
`new_vocab`<a id="new_vocab"></a>
</td>
<td>
A `namedtuple` alias for field number 0
</td>
</tr><tr>
<td>
`new_vocab_size`<a id="new_vocab_size"></a>
</td>
<td>
A `namedtuple` alias for field number 1
</td>
</tr><tr>
<td>
`num_oov_buckets`<a id="num_oov_buckets"></a>
</td>
<td>
A `namedtuple` alias for field number 2
</td>
</tr><tr>
<td>
`old_vocab`<a id="old_vocab"></a>
</td>
<td>
A `namedtuple` alias for field number 3
</td>
</tr><tr>
<td>
`old_vocab_size`<a id="old_vocab_size"></a>
</td>
<td>
A `namedtuple` alias for field number 4
</td>
</tr><tr>
<td>
`backup_initializer`<a id="backup_initializer"></a>
</td>
<td>
A `namedtuple` alias for field number 5
</td>
</tr><tr>
<td>
`axis`<a id="axis"></a>
</td>
<td>
A `namedtuple` alias for field number 6
</td>
</tr>
</table>
© 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/compat/v1/train/VocabInfo