/multi-level-product-classification

An ML model to classify products in to a hierarchy of categories

Primary LanguageJupyter Notebook

multi-level-product-classification

[work-in-progress][readme-in-progress!]

An ML model to classify products in to a hierarchy of categories.

Steps

Execute the notebooks in the order tokenize.ipynb, classify.ipynb, predict.ipynb

TODO

  • Refactor the code
  • Add sample training/testing data
  • Support n levels (dynamic)