Downloads · 30 days
17
2% of all-time downloads
kaggle-comp-test
Crasher222/kaggle-comp-test
kaggle-comp-test is a text classification model from Crasher222. 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: 25805800
All-time downloads
915
Public
Repo size
2.7 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin1.3 GB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Oct 24, 2021
- SHA
- 8a1b24c9
Try a prompt
Model card
From the Hugging Face model README
Model Finetuned from BERT-base for
- Problem type: Multi-class Classification
- Model ID: 25805800
Validation Metrics
- Loss: 0.4422711133956909
- Accuracy: 0.8615328555811976
- Macro F1: 0.8642434650461513
- Micro F1: 0.8615328555811976
- Weighted F1: 0.8617743626671308
- Macro Precision: 0.8649112225076049
- Micro Precision: 0.8615328555811976
- Weighted Precision: 0.8625407179375096
- Macro Recall: 0.8640777539828228
- Micro Recall: 0.8615328555811976
- Weighted Recall: 0.8615328555811976
Usage
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("Crasher222/kaggle-comp-test")
tokenizer = AutoTokenizer.from_pretrained("Crasher222/kaggle-comp-test")
inputs = tokenizer("I am in love with you", return_tensors="pt")
outputs = model(**inputs)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Oct 24, 2021