tf.contrib.framework.load_embedding_initializer

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

tf.contrib.framework.load_embedding_initializer(
    ckpt_path, embedding_tensor_name, new_vocab_size, embedding_dim, old_vocab_file,
    new_vocab_file, old_vocab_size=-1, num_oov_buckets=0, initializer=None,
    max_rows_in_memory=-1
)

Обёртка вокруг load_and_remap_matrix_initializer() , специализированная для загрузки весов вложений и перепривязки согласно предоставленным файлам словарей. См. документацию по load_and_remap_matrix_initializer() для получения более подробной информации.

Примечание: Только для использования с переменными/словарями, разделяемыми по разделам.

Аргументы
`ckpt_path`	Путь к контрольной точке TensorFlow (версия 2, `TensorBundle`) из которой будет загружена старая матрица `Tensor` .
`embedding_tensor_name`	Имя 2-мерного `Tensor` для загрузки из контрольной точки.
`new_vocab_size`	Количество записей в новом словаре.
`embedding_dim`	`int` , определяющее размер векторов вложений из контрольной точки. Должно соответствовать количеству столбцов в старой матрице вложений.
`old_vocab_file`	Скалярный `Tensor` типа `string` , содержащий путь к файлу старого словаря.
`new_vocab_file`	Скалярный `Tensor` типа `string` , содержащий путь к файлу нового словаря.
`old_vocab_size`	Количество записей для рассмотрения в старом словаре. С использованием значения по умолчанию -1 будет использоваться весь старый файл словаря строк. В противном случае будут рассмотрены только первые `old_vocab_size` записи для перепривязки. Должно быть меньше длины `old_row_vocab_file`.
`num_oov_buckets`	`int` , определяющее количество ведер вне словаря для использования. Должно быть >= 0.
`initializer`	Функция инициализации, которая принимает 1-мерный тензор в качестве аргумента для указания формы возвращаемого тензора. Если `None`, по умолчанию используется `truncated_normal_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_embedding_initializer