Hugging Face's logo

neil-code
/
autotrain-text-classific-imdb

Text Classification
Transformers
Safetensors
bert
Trained with AutoTrain
text-embeddings-inference
Model card Files
xet
 Community
How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="neil-code/autotrain-text-classific-imdb")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("neil-code/autotrain-text-classific-imdb")
model = AutoModelForSequenceClassification.from_pretrained("neil-code/autotrain-text-classific-imdb")
Quick Links

Model Trained Using AutoTrain

  • Problem type: Text Classification

Validation Metrics

loss: 0.32178911566734314

f1: 0.8431746031746032

precision: 0.8952808988764045

recall: 0.7968

auc: 0.94403024

accuracy: 0.8518

Downloads last month
5
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
Text Classification
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support