Tensorflow:如何从训练有素的模型中获取哈希表

时间:2018-04-06 18:28:49

标签: tensorflow hashmap hashtable

我使用tensorflow训练了一个LinearClassifier。为了处理分类值,我使用类似的方法this link使用:

tf.feature_column.categorical_column_with_vocabulary_list, tf.feature_column.categorical_column_with_hash_bucket

我也使用了cross_columns,如链接提供的

我可以很容易地获得系数 。例如,如果我想获取功能'设备':

的权重

model.get_variable_value('linear/linear_model/device/')

我也可以轻松地将这些系数映射到使用tf.feature_column.categorical_column_with_vocabulary_list训练的分类变量值。但是,对于使用tf.feature_column.categorical_column_with_hash_bucket训练的分类变量值,这是不可能的。我无法得到如何进行映射。由于哈希表是动态完成的,我不知道哪个权重对应哪个值。

我已经查看了几个地方甚至深入研究estimator

的代码

有什么想法吗?

0 个答案:

没有答案