python - TensorFlow 无法将字符串转换为数字

Question

我正在构建一个神经网络，它应该以某种方式对输入的单词进行分类。在不深入了解网络本身的情况下，我一直在寻找一种方法将我的输入单词转换为整数格式，以便使用 TensorFlow 的 tf.nn.embedding_lookup(...) 进行输入编码。

我注意到 tf.string_to_number() 存在，所以我尝试使用它，但失败了。首先我认为这与我在我的网络中所做的事情有关，但即使在做类似

import tensorflow as tf
s = tf.string_to_number("TEST", out_type=tf.int32)
sess = tf.InteractiveSession()
sess.run(s)

在 python 控制台中，我得到了同样的错误

tensorflow.python.framework.errors.InvalidArgumentError: 
StringToNumberOp could not correctly convert string: TEST

我还尝试先创建一个 tf.constant("TEST", dtype=tf.string) 并将其传递给 tf.string_to_number() 并运行它在网络服务器上测试代码以确保它与我的设置无关，但结果相同。

谁能告诉我我在这里缺少什么？提前致谢!

Answer 1

Can anyone tell me what I'm missing here?

您错过了 string_to_number 的目的，它应该将表示为字符串的数字转换为数字类型，例如 tf.string_to_number('1') ，它不是字符串的“一个热编码器”(它如何能够首先计算出词汇中的大小？)

tensorflow 本身有一个很好的教程，它展示了如何在 word2vec_basic.py 中训练嵌入模型它贯穿所有内容，从数据读取开始，到使用查找操作完全嵌入结束。