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