SeyedAli/Persian-Text-Emotion
Viewer • Updated • 6.95k • 53 • 4
How to use NLPclass/parsBERT_text_emotion_classification with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="NLPclass/parsBERT_text_emotion_classification") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("NLPclass/parsBERT_text_emotion_classification")
model = AutoModelForSequenceClassification.from_pretrained("NLPclass/parsBERT_text_emotion_classification")using ParsBert for emotion text classification in persian dataSet
This model is designed to emotion text classification in persian dataSet, which uses the pre-trained model ParsBert in persian language.
this model train with 5558 text from SeyedAli/Persian-Text-Emotion dataSet
this model test with 1399 text from SeyedAli/Persian-Text-Emotion dataSet with 6 class