# How to implement hierarchical labeling classification?

I am currently working on the task of eCommerce product name classification, so I have categories and subcategories in product data. I noticed that using subcategories as labels delivers worse results (84% acc) than categories (94% acc). But subcategories are more precise as labels, what's important for the whole task. And then I got an idea to first do category classification and then based on the results continue with subcategories within the predicted category.

The problem here is that I do not know how to approach this problem/define network architecture. Any hints on the neural networks, how to deal with it?

Currently I defined network like this:

model = Sequential()

• Did you manage to solve this question somehow? – Franco Piccolo Jun 24 '20 at 11:25
• Unfortunately, not yet. – chacid Jun 25 '20 at 12:08

For this problem, you need to develop three models:

Model 1- for two main categories
Model 2- for sub-category A
Model 3- For sub-category B


So when you want to predict the result for an unseen data, first you use the Model 1, to find the main category. Based on the prediction and by using an if-else statement, you decide to perform another prediction using Model 2 or Model 3.

Consequently, the Model 1 is a binary classification task, but Model 2&3 are mutli-class classification task. Your networks may look like this:

Model 1:

model1 = Sequential()
...


Model 2 & 3:

model2 = Sequential()