张量流模型发生器

Question

我正在制作一个移动应用程序，它使用TensorFlow Lite对推文进行文本分类。我已经用TensorFlow示例模型成功地实现了这一点，但该模型是在IMDB电影评论上训练的，我希望在设备上有一个自定义模型，该模型是在推特上训练的，以提高准确性。我有这个领域的培训和测试集，我正试图在本教程之后创建一个自定义模型https://www.tensorflow.org/lite/tutorials/model_maker_text_classification

不过，我遇到了一个pythionKeyError，不知道为什么。这是一个屏幕截图

您可以在右侧看到我的csv示例，我使用TextClassifierDataLoader有一个label和Sentence列。我不明白为什么会发生这个关键错误，我解释它，因为它找不到一个名为“句子”的专栏，但很清楚它就在那里

有什么想法吗

Answer 1

问题是“句子”一栏开头有一个空格

Answer 2

在我看来，is_training在课堂上不是一个可用的参数，所以如果你想编译它，就去掉那一行/参数（它本来应该做什么，我还没有研究过）

https://github.com/tensorflow/examples/blob/1dc6978e2141e7a5efebcf6971b3afa9cb055679/tensorflow_examples/lite/model_maker/core/data_util/text_dataloader.py#L90