我用多个级别的因子变量(例如,邮政编码等)训练了一个模型。我使用了 tf.feature_column.categorical_column_with_hash_bucket 和 tf.feature_column.embedding_column 函数:
categorical_column_zip =
tf.feature_column.categorical_column_with_hash_bucket('zip_code',
_HASH_BUCKET_SIZE)
categorical_feature_zip_emb =
tf.feature_column.embedding_column(categorical_column =
categorical_column_zip, dimension = 6)
现在,我需要探索嵌入(尤其是可视化)。具体来说,我需要访问一些邮政编码的嵌入内容。看来可以用
完成zip_embeddings = tf.get_variable('categorical_feature_zip_emb',
[n_distinct_zip_codes, embedding_size])
embedded_zip_ids = tf.nn.embedding_lookup(zip_embeddings,
zip_ids)
我需要感兴趣的邮政编码列表中的zip_id。
问题:我如何获取映射{邮政编码值:邮政编码索引}
tf.feature_column.embedding_column 使用的是什么?