Spec-Zone.ru › TensorFlow

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

Spec-Zone.ru

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