Spec-Zone.ru › TensorFlow 1.15

tf.contrib.layers.shared_embedding_columns

Создаёт список _EmbeddingColumn с общим вложением.

tf.contrib.layers.shared_embedding_columns(
    sparse_id_columns, dimension, combiner='mean', shared_embedding_name=None,
    initializer=None, ckpt_to_load_from=None, tensor_name_in_ckpt=None,
    max_norm=None, trainable=True
)
Аргументы
sparse_id_columns Итерируемый список _SparseColumn, например, созданный функциями sparse_column_with_* или crossed_column. Обратите внимание, что combiner , определённые в каждом sparse_id_column, игнорируются.
dimension Целое число, определяющее размерность вложения.
combiner Строка, определяющая способ сокращения, если в одной строке есть несколько записей. В настоящее время поддерживаются "mean", "sqrtn" и "sum", по умолчанию используется "mean". "sqrtn" часто обеспечивает хорошую точность, особенно со столбцами bag-of-words. Каждый из этих вариантов можно рассматривать как нормализацию на уровне примера для столбца:
  • "sum": не нормализовать
  • "mean": нормализация l1
  • "sqrtn": нормализация l2 Для получения дополнительной информации: tf.embedding_lookup_sparse.
shared_embedding_name (Необязательно). Строка, определяющая имя общих весов вложения. Это потребуется, если вы хотите сослаться на общее вложение отдельно от сгенерированных _EmbeddingColumn.
initializer Функция инициализации переменной, которая будет использоваться при инициализации переменной вложения. Если не указано, по умолчанию используется tf.compat.v1.truncated_normal_initializer со средним значением 0,0 и стандартным отклонением 1/sqrt(sparse_id_columns[0].length).
ckpt_to_load_from (Необязательно). Строка, представляющая имя/шаблон контрольной точки для восстановления весов столбца. Требуется, если tensor_name_in_ckpt не равно None.
tensor_name_in_ckpt (Необязательно). Имя Tensor в предоставленной контрольной точке, из которой следует восстановить веса столбца. Требуется, если ckpt_to_load_from не равно None.
max_norm (Необязательно). Если не None, значения вложения нормализуются по l2 до значения max_norm.
trainable (Необязательно). Должно ли вложение обучаться. По умолчанию True
Возвращает
Кортеж _EmbeddingColumn с общим пространством вложения.
Возможные исключения
ValueError если sparse_id_columns пустой или его элементы несовместимы друг с другом.
TypeError если sparse_id_columns не является последовательностью или является строкой. Если хотя бы один элемент sparse_id_columns не является SparseColumn или WeightedSparseColumn.

© 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/layers/shared_embedding_columns

Spec-Zone.ru

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