Skip to content

jaswant-rajput/classification-finetuned-model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

text-classification-finetuned-model

Finetuned a distilbert model on emotions dataset from huggingface and uploaded the dataset on huggingfacehub

model_id on huggingface = "jaswant50/distilbert-base-uncased-finetuned-emotion"

Use a pipeline as a high-level helper

from transformers import pipeline

pipe = pipeline("text-classification", model="jaswant50/distilbert-base-uncased-finetuned-emotion")

Load model directly

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("jaswant50/distilbert-base-uncased-finetuned-emotion") model = AutoModelForSequenceClassification.from_pretrained("jaswant50/distilbert-base-uncased-finetuned-emotion")

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published