张量流模型制作器

Question

我正在制作一个使用TensorFlow Lite对推文执行文本分类的移动应用程序。我已经使用TensorFlow示例模型成功做到了这一点，但是该模型在IMDB电影评论中得到了训练，并且我希望在设备上拥有一个经过推文训练的自定义模型，以提高准确性。我为此领域提供了培训和测试集，并按照本教程https://www.tensorflow.org/lite/tutorials/model_maker_text_classification

尝试创建自定义模型

虽然我遇到了一个KeyError的pythion，却不知道为什么。这是一个屏幕截图

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

有什么想法吗？

Answer 1

在我看来is_training不是该类中的可用参数，因此，如果要对其进行编译，只需取出该行/参数（首先应该做的是，我没有研究过）

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

Answer 2

问题是“句子”列的开头是空的