如何将字符串标签打包到savedModel中

Question

我有字符串标签，例如＆＃34; cat＆＃34;，＆＃34; dog＆＃34;。我可以直接将字符串标签提供给Tensorflow中的深度学习模型，并将字符串标签作为预测吗？我正在寻找相当于sklearn的labelEncoder sklearn.preprocessing import LabelEncoder

如果无法做到这一点，有没有办法将标签打包到savedModel protobuf文件中并在服务时间内根据索引检索它们？我正在使用Estimator的export_savedModel API。 assets_extra是正确的方式吗？ https://github.com/tensorflow/serving/issues/55上的那个不使用savedModel格式。

Answer 1

在深度学习中处理标签数据的典型方法是将标签嵌入向量空间。语言模型使用word embedding定期执行。 TensorFlow提供可用于您的embedding lookup操作。