首页 > 解决方案 > 我应该使用哪种型号?- 多标签分类

问题描述

我是数据科学的新手,所以我的问题可能是基本的。

我有一个数据集。第一列是人们对问题的评论(作为文本),第二列是该失败的类别/标签(作为文本)。我的第二列有很多失败类型。

我想训练一个模型。当输入另一个评论并解释问题时,模型应该对故障进行分类。

我可以使用 Keras Sequential 模型吗?或者我应该使用不同的模型?如果您可以分享与我的问题相关的链接,我将不胜感激。

标签: kerasdatasetdata-sciencemultilabel-classification

解决方案


您肯定可以使用 Keras Sequential 模型。现在作为初学者,尝试使用密集层,您也可以使用卷积神经网络......顺便说一句,尝试使用 tensorflow.keras.preprocessing.text 标记器将每个单词标记为数字,以便机器可以理解。

有关更多信息,请在 Google 上搜索文本分类并搜索 Tokenizer。


推荐阅读