如何在TensorFlow的分类列中获取索引映射的键

时间:2018-11-15 17:15:31

标签: python-3.x tensorflow

我用多个级别的因子变量(例如,邮政编码等)训练了一个模型。我使用了 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 使用的是什么?

0 个答案:

没有答案