tensorflow - 在 Keras 模型中使用 Tensorflow feature_column

Question

Tensorflow feature_column 如何与 Keras 模型结合使用？

例如对于 Tensorflow 估计器，我们可以使用 Tensorflow Hub 中的嵌入列:

embedded_text_feature_column = hub.text_embedding_column(
    key="sentence", 
    module_spec="https://tfhub.dev/google/nnlm-en-dim128/1")

estimator = tf.estimator.DNNClassifier(
hidden_units=[100],
feature_columns=[embedded_text_feature_column],
n_classes=2,
optimizer=tf.train.AdamOptimizer(learning_rate=0.001))

但是，我想使用 TF Hub text_embedding_column 作为 Keras 模型的输入。例如

net = tf.keras.layers.Input(...) # use embedding column here
net = tf.keras.layers.Flatten()
net = Dense(100, activation='relu')(net)
net = Dense(2)(net)

这可能吗？

Answer 1

答案似乎是您不使用特征列。 Keras 附带了自己的一组图像和文本预处理函数，因此您可以使用它们。

所以基本上 tf.feature_columns 是为高级 API 保留的。然后，tf.keras.preprocessing() 函数与 tf.keras 模型一起使用。

这里是 keras 文档中有关预处理数据部分的链接。 https://keras.io/preprocessing/text/

这是另一篇 Stackoverflow 帖子，其中包含此方法的示例。

Add Tensorflow pre-processing to existing Keras model (for use in Tensorflow Serving)