Creating an E-Commerce Product Category Classifier using Deep Learning — Part 2

Problem Description :

Machine Learning Pipeline :

Fig 1. The machine learning pipeline for the category prediction task.

Data Splitting

Fig 2. A mathematical explanation of TF-IDF

Binary Classification Technique :

Fig 3. Snippet showing individual accuracy/ROC of individual product classification model and finally accuracy/ROC of the overall model.
Fig 4. Predictions on a few samples, we can observe how it predicts Samsung as cell phone, Duracell battery as houseware.

Deep Learning-Based Models :

Fig 5. Simple neural net-based multi-classification product category prediction model
Fig 6. Performance tracking for neural net model
Fig 7. Prediction by the neural net multi-classification model, we can observe how it has classified Kung Fu Panda as a Video Game which is quite well.
Fig 8. Conv neural net-based multi-classification product category prediction model
Fig 9. Performance tracking for conv neural net model
Fig 10. LSTM + Glove neural net-based multi-classification product category prediction model
Fig 11. Performance tracking for LSTM + Glove neural net model
GET /product/categoryPrediction
{
"name":"product name",
"description":"product description"
}
Fig 12. Final prediction from LSTM+Glove based model as a product category classifier.

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store