Downloads · 30 days
8
1% of all-time downloads
autonlp-department-classification-534915130
rodrigogelacio/autonlp-department-classification-534915130
autonlp-department-classification-534915130 is a text classification model from rodrigogelacio. Use it when you need a label for a piece of text. It is set up for transformers.
- Problem type: Multi-class Classification - Model ID: 534915130 - CO2 Emissions (in grams): 1.4862856774320061
All-time downloads
771
Public
Repo size
873 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin436 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jan 28, 2022
- SHA
- fba721a4
Try a prompt
Model card
From the Hugging Face model README
Model Trained Using AutoNLP
- Problem type: Multi-class Classification
- Model ID: 534915130
- CO2 Emissions (in grams): 1.4862856774320061
Validation Metrics
- Loss: 0.37066277861595154
- Accuracy: 0.9204545454545454
- Macro F1: 0.9103715740678612
- Micro F1: 0.9204545454545455
- Weighted F1: 0.9196871607509906
- Macro Precision: 0.9207759152612094
- Micro Precision: 0.9204545454545454
- Weighted Precision: 0.922177301864802
- Macro Recall: 0.9055002187355129
- Micro Recall: 0.9204545454545454
- Weighted Recall: 0.9204545454545454
Usage
You can use cURL to access this model:
$ curl -X POST -H "Authorization: Bearer YOUR_API_KEY" -H "Content-Type: application/json" -d '{"inputs": "I love AutoNLP"}' https://api-inference.huggingface.co/models/rodrigogelacio/autonlp-department-classification-534915130
Or Python API:
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("rodrigogelacio/autonlp-department-classification-534915130", use_auth_token=True)
tokenizer = AutoTokenizer.from_pretrained("rodrigogelacio/autonlp-department-classification-534915130", use_auth_token=True)
inputs = tokenizer("I love AutoNLP", return_tensors="pt")
outputs = model(**inputs)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- unk
- Created
- Mar 2, 2022
- Updated
- Jan 28, 2022