Modelpublic

tradero/distilbert-user-intent

sourceHugging Faceupdated 3y agoView on Hugging Face

0likes8downloads

Model Card

Model Trained Using AutoTrain

—Problem type: Multi-class Classification
—Model ID: 49241119078
—CO2 Emissions (in grams): 0.4690

Validation Metrics

—Loss: 1.676
—Accuracy: 0.800
—Macro F1: 0.733
—Micro F1: 0.800
—Weighted F1: 0.733
—Macro Precision: 0.700
—Micro Precision: 0.800
—Weighted Precision: 0.700
—Macro Recall: 0.800
—Micro Recall: 0.800
—Weighted Recall: 0.800

Usage

Python API:

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained("tradero/distilbert-user-intent", use_auth_token=True)

tokenizer = AutoTokenizer.from_pretrained("tradero/distilbert-user-intent", use_auth_token=True)

inputs = tokenizer("I love AutoTrain", return_tensors="pt")

outputs = model(**inputs)