Ayad0net / dl-product-category-classification

Deep Learning Model for Product Category Classification

💡 Note: the original repo is: 🔗 https://github.com/yang-zhang/product_category

📦 Installation

1. clone the repo ...

git clone https://github.com/Ayad0net/dl-product-category-classification.git

2. install packages ...

Inside the repo directory ...

pip install -r requirements.txt --no-cache-dir

3. download pretrained model...

Inside the repo directory ...

wget https://github.com/yang-zhang/product_category/releases/download/v0.0.1/transformer_20210307D3.ckpt -P data

Example

python3 predict.py -t "Lykmera Famous TikTok Leggings, High Waist Yoga Pants for Women, Booty Bubble Butt Lifting Workout Running Tights"

output

Sports & Outdoors: 0.997
Sports & Fitness: 0.994
Exercise & Fitness: 0.980
Clothing: 0.961
Yoga: 0.905