Downloads · 30 days
316
0% of all-time downloads
industry-classification
sampathkethineedi/industry-classification
industry-classification is a text classification model from sampathkethineedi. Use it when you need a label for a piece of text. It is set up for transformers.
DistilBERT Model to classify a business description into one of 62 industry tags. Trained on 7000 samples of Business Descriptions and associated labels of companies in India.
All-time downloads
220K
Public
Repo size
804 MB
Likes
24
Public
Repo makeup
Click a slice to open those files.
.h5268 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- distilbert
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jul 16, 2020
- SHA
- 99142320
Try a prompt
Model card
From the Hugging Face model README
industry-classification
Model description
DistilBERT Model to classify a business description into one of 62 industry tags. Trained on 7000 samples of Business Descriptions and associated labels of companies in India.
How to use
PyTorch and TF models available
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
tokenizer = AutoTokenizer.from_pretrained("sampathkethineedi/industry-classification")
model = AutoModelForSequenceClassification.from_pretrained("sampathkethineedi/industry-classification")
industry_tags = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
industry_tags("Stellar Capital Services Limited is an India-based non-banking financial company ... loan against property, management consultancy, personal loans and unsecured loans.")
'''Ouput'''
[{'label': 'Consumer Finance', 'score': 0.9841355681419373}]
Limitations and bias
Training data is only for Indian companies
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Jul 16, 2020