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
)

Примеры использования VocabInfo для запуска с предварительной инициализацией см. в tf.estimator.WarmStartSettings.

Аргументы: 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