Multi-level Product Category Prediction through Text Classification

arXiv:2403.01638v1 Announce Type: new
Abstract: This article investigates applying advanced machine learning models, specifically LSTM and BERT, for text classification to predict multiple categories in the retail sector. The study demonstrates how applying data augmentation techniques and the focal loss function can significantly enhance accuracy in classifying products into multiple categories using a robust Brazilian retail dataset. The LSTM model, enriched with Brazilian word embedding, and BERT, known for its effectiveness in understanding complex contexts, were adapted and optimized for …

abstract accuracy advanced article arxiv augmentation bert classification cs.cl data function loss lstm machine machine learning machine learning models multiple prediction product products retail sector study text text classification through type