您是否尝试过NLTK，您正在寻找的是本书的第 6 章。基本上你需要做的是：

• 标记用户输入。
• 从标记化的单词中提取向量集。
• 使用一些给定的文本和相同的向量集训练您的模型。

您可以使用您的模型对文档进行分类。

另一个建议是，您可以使用输入中的每个单词来使用大型语料库的训练集来评估某个类别，而不是提取向量集，您可以确定它包含所有单词。然后你乘以每个单词在一个类别中的概率来决定文档所属的位置。