使用tf-transfrom将单词预测ID转换为字符串

时间:2019-05-08 07:20:29

标签: tensorflow-transform

我正在使用带有(tf 1.13版本)的tensorflow-transfrom来训练一个命名的实体识别器。我有words功能作为输入,有tags作为标签。 tft.compute_and_apply_vocabulary将它们都转换为整数ID,但是也有必要能够将预测ID转换回字符串,但是我在文档或示例中找不到有效的方法。此转换是否有任何反向功能,或者应该怎么做?

我知道tf.contrib.lookup中的查找实用程序功能,这可能是解决方案,但是我不熟悉它们与tft.compute_and_apply_vocabulary中使用的方法的兼容性(例如,将vocab序列化为文件时)通过tft.compute_and_apply_vocabulary进行维护的令牌顺序)

0 个答案:

没有答案