来自一个张量流特征的两组共享嵌入？

Question

如何从相同的 tensorflow 特征列创建两组共享嵌入？

这个小例子

import tensorflow as tf

data = {"A": [0, 1, 2, 3], "B": [2, 1, 0, 3]}


def add_label(example):
    return example, 1


def input_fn():
    dset = tf.data.Dataset.from_tensor_slices(data).map(add_label).batch(2)
    return dset


def model_fn(features, labels, mode, params):
    colA = tf.feature_column.categorical_column_with_vocabulary_list("A", [0, 1, 2, 3])
    colB = tf.feature_column.categorical_column_with_vocabulary_list("B", [0, 1, 2, 3])

    model1_embedddings = tf.feature_column.shared_embeddings(categorical_columns=[colA, colB], dimension=2)
    X1 = tf.keras.layers.DenseFeatures(model1_embedddings)(features)
    output1_output = tf.reduce_sum(X1, axis=1)

    with tf.compat.v1.variable_scope("other", reuse=False):
        model2_embedddings = tf.feature_column.shared_embeddings(categorical_columns=[colA, colB], dimension=2)
    X2 = tf.keras.layers.DenseFeatures(model2_embedddings)(features)
    output2_output = tf.reduce_sum(X2, axis=1)

    loss = tf.losses.mean_squared_error(labels, output1_output + output2_output)

    optimizer = tf.compat.v1.train.AdamOptimizer(learning_rate=0.01)
    train_op = optimizer.minimize(loss=loss)
    return tf.estimator.EstimatorSpec(mode=mode, loss=loss, train_op=train_op)


estimator = tf.estimator.Estimator(model_fn=model_fn)
estimator.train(input_fn=input_fn)

崩溃

ValueError: Variable A_B_shared_embedding already exists, disallowed. Did you mean to set reuse=True or reuse=tf.AUTO_REUSE in VarScope?

似乎应该可以使用 variable_scope 或 name_scope 来让它工作，但到目前为止运气不好。

Answer 1

shared_embeddings 有一个选项可以设置新的嵌入集合名称

model2_embedddings = tf.feature_column.shared_embeddings(
 categorical_columns=[colA, colB], dimension=2,
 shared_embedding_collection_name="other")

来自一个张量流特征的两组共享嵌入？

Two sets of shared embeddings from one tensorflow feature?

tensorflow

tensorflow-estimator

tf.keras