多层多类别分类

时间:2019-06-11 13:22:42

标签: python-3.x keras lstm recurrent-neural-network multiclass-classification

我有一个电子商务类别结构,其中每个父类别都可以有子类别。例如:

父类别1->子类别1->子类别2->子类别3
父类别2->子类别1->子类别2

我也拥有电子商务网站中的所有搜索关键字。

我的问题是为搜索查询分配正确的子类别层次结构

所需的输入和输出如下:

输入:

搜索关键字; L1类,L2类,L3类

办公椅;家具;椅子和座椅;办公椅
惠普笔记本电脑;技术;计算机与笔记本电脑;笔记本电脑
苹果ipad;技术;平板电脑
深色咖啡;休息室;咖啡

如果最终用户在电子商务网站上输入的搜索关键字是:

台式机监控器
淡咖啡

然后输出文件应包含正确的搜索关键字分类,如下所示:

台式机显示器;技术;计算机和笔记本电脑
淡咖啡;休息室;咖啡

我已经尝试通过使用LSTM和使用keras进行单词嵌入来对父类别进行多类分类,并且能够分配

办公椅->家具

让我知道是否有人在处理类似问题。

1 个答案:

答案 0 :(得分:0)

我建议为层次结构树中每个节点的子级训练一个分类器。例如,分类器编号1将父类别1和父类别2之间的项目分类(2类分类)。然后训练带有父类别1的项的分类器,该项在subcategory1,subcategory2和suncategory3(3类分类)之间的项之间进行分类。等等。