Amir-mjafari/ecommerce-classification
I used Keras library to train a Convolutional Neural Network Model to classify 21627 noisy images of e-commerce products into 27 categories. Due to the availability of corresponding text descriptions and categorical features, I implemented a Transformer Model (pre-trained BERT) with PyTorch library to process the text data. The ensemble of the two mentioned models achieved 96 percent accuracy.
No commits in the last 6 months.
Stars
—
Forks
—
Language
Jupyter Notebook
License
Unlicense
Category
Last pushed
Oct 15, 2021
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/transformers/Amir-mjafari/ecommerce-classification"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
lonePatient/Bert-Multi-Label-Text-Classification
This repo contains a PyTorch implementation of a pretrained BERT model for multi-label text...
mim-solutions/bert_for_longer_texts
BERT classification model for processing texts longer than 512 tokens. Text is first divided...
OctoberChang/X-Transformer
X-Transformer: Taming Pretrained Transformers for eXtreme Multi-label Text Classification
QData/LaMP
ECML 2019: Graph Neural Networks for Multi-Label Classification
illiterate/BertClassifier
基于PyTorch的BERT中文文本分类模型（BERT Chinese text classification model implemented by PyTorch）