tf.contrib.layers.shared_embedding_columns
Stay organized with collections
Save and categorize content based on your preferences.
Creates a list of _EmbeddingColumn
sharing the same embedding.
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
)
Args |
sparse_id_columns
|
An iterable of _SparseColumn , such as those created by
sparse_column_with_* or crossed_column functions. Note that combiner
defined in each sparse_id_column is ignored.
|
dimension
|
An integer specifying dimension of the embedding.
|
combiner
|
A string specifying how to reduce if there are multiple entries in
a single row. Currently "mean", "sqrtn" and "sum" are supported, with
"mean" the default. "sqrtn" often achieves good accuracy, in particular
with bag-of-words columns. Each of this can be thought as example level
normalizations on the column:
- "sum": do not normalize
- "mean": do l1 normalization
- "sqrtn": do l2 normalization
For more information:
tf.embedding_lookup_sparse .
|
shared_embedding_name
|
(Optional). A string specifying the name of shared
embedding weights. This will be needed if you want to reference the shared
embedding separately from the generated _EmbeddingColumn .
|
initializer
|
A variable initializer function to be used in embedding
variable initialization. If not specified, defaults to
tf.compat.v1.truncated_normal_initializer with mean 0.0 and standard
deviation 1/sqrt(sparse_id_columns[0].length).
|
ckpt_to_load_from
|
(Optional). String representing checkpoint name/pattern
to restore the column weights. Required if tensor_name_in_ckpt is not
None.
|
tensor_name_in_ckpt
|
(Optional). Name of the Tensor in the provided
checkpoint from which to restore the column weights. Required if
ckpt_to_load_from is not None.
|
max_norm
|
(Optional). If not None, embedding values are l2-normalized to the
value of max_norm.
|
trainable
|
(Optional). Should the embedding be trainable. Default is True
|
Returns |
A tuple of _EmbeddingColumn with shared embedding space.
|
Raises |
ValueError
|
if sparse_id_columns is empty, or its elements are not
compatible with each other.
|
TypeError
|
if sparse_id_columns is not a sequence or is a string. If at
least one element of sparse_id_columns is not a SparseColumn or a
WeightedSparseColumn .
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2020-10-01 UTC."],[],[]]