Automatic multi-class classification models

For the e-commerce marketplace FlipKart :

1. Study the feasability of an automatic multi-class classification model, to tidy the product into the right categories from their image and/or description only.
2. Benchmark
3. Implement the right model and use data augmentation to improve the results

The data was collected from the open-source platform Flipkart dataset.

The code is available here :

• collect, clean & feasability study
• Benchmark for classification models
• Data augmentation for the selected model

Download PDF Presentation