Spec-Zone.ru › TensorFlow 1.15

tf.contrib.framework.load_linear_multiclass_bias_initializer

Загружает предварительно обученные смещения многоклассовой модели линейных моделей из контрольной точки.

tf.contrib.framework.load_linear_multiclass_bias_initializer(
    ckpt_path, bias_tensor_name, new_class_vocab_size, old_class_vocab_file,
    new_class_vocab_file, num_class_oov_buckets=0, initializer=None,
    max_rows_in_memory=-1
)

Обёртка вокруг load_and_remap_matrix_initializer() специализированная для загрузки смещений многоклассовой модели и переназначения в соответствии с предоставленными файлами словаря. Смотрите документацию для load_and_remap_matrix_initializer() для получения более подробной информации. В этом случае предоставленный row_vocab — словарь классов, а ожидаемая форма — [new_class_vocab_size, 1].

Аргументы
ckpt_path Путь к контрольной точке TensorFlow (версия 2, TensorBundle) из которой будет загружена старая матрица Tensor.
bias_tensor_name Имя тензора для загрузки из контрольных точек.
new_class_vocab_size Количество элементов в новом словаре классов.
old_class_vocab_file Скаляр Tensor типа string, содержащий путь к старому файлу словаря классов.
new_class_vocab_file Скаляр Tensor типа string, содержащий путь к новому файлу словаря классов.
num_class_oov_buckets int , определяющий количество ведер вне словаря для использования для классов. Должно быть >= 0.
initializer Функция инициализации, которая принимает 1-мерный тензор в качестве аргумента для указания формы возвращаемого тензора. Если None, по умолчанию используется zeros_initializer().
max_rows_in_memory int , определяющий максимальное количество строк для загрузки из контрольной точки за раз. Если меньше или равно 0, вся матрица будет загружена в память. Установка этого аргумента меняет повышенные чтения с диска на меньшее использование памяти.
Возвращаемое значение
Функция инициализации переменной.

© 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/framework/load_linear_multiclass_bias_initializer

Spec-Zone.ru

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