PyTorch Implementation of Deep Hierarchical Classification for Category Prediction in E-commerce System

商品类目预测，使用 Spring Boot 开发框架和 Spark MLlib 机器学习框架，通过 TF-IDF 和 Bayes 算法，训练出一个商品类目预测模型。该模型可以根据商品名称自动预测出商品类目。项目对外提供 RESTFul 接口。

Implementation for "AttentionXML: Label Tree-based Attention-Aware Deep Model for High-Performance Extreme Multi-Label Text Classification"

Multi-Class Text Classification for products based on their description with Machine Learning algorithms and Neural Networks (MLP, CNN, Distilbert).

A Natural Language Processing based project that aims at categorizing e-commerce products based on their description using Machine Learning and Deep Learning techniques (implemented using PyTorch).