Update README.md
Browse files
README.md
CHANGED
@@ -7,7 +7,7 @@ This model is a fine-tuned version of the Longformer base model for sentiment an
|
|
7 |
- Model Type: Longformer
|
8 |
- Task: Sentiment Analysis
|
9 |
- Training Data: 4000 customer support tickets (Approx 1000 for each class)
|
10 |
-
- Number of Parameters:
|
11 |
|
12 |
## Performance
|
13 |
- Overall Accuracy: 74.84%
|
@@ -32,6 +32,7 @@ weighted avg 0.76 0.75 0.73 477
|
|
32 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
33 |
import torch
|
34 |
|
|
|
35 |
model = AutoModelForSequenceClassification.from_pretrained("{hf_model_name}")
|
36 |
tokenizer = AutoTokenizer.from_pretrained("{hf_model_name}")
|
37 |
|
@@ -48,4 +49,4 @@ print(f"Predicted sentiment: {sentiment_map[prediction]}")
|
|
48 |
|
49 |
|
50 |
## License
|
51 |
-
|
|
|
7 |
- Model Type: Longformer
|
8 |
- Task: Sentiment Analysis
|
9 |
- Training Data: 4000 customer support tickets (Approx 1000 for each class)
|
10 |
+
- Number of Parameters: 149M
|
11 |
|
12 |
## Performance
|
13 |
- Overall Accuracy: 74.84%
|
|
|
32 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
33 |
import torch
|
34 |
|
35 |
+
hf_model_name= 'Muddassar/longformer-base-sentiment-5-classes'
|
36 |
model = AutoModelForSequenceClassification.from_pretrained("{hf_model_name}")
|
37 |
tokenizer = AutoTokenizer.from_pretrained("{hf_model_name}")
|
38 |
|
|
|
49 |
|
50 |
|
51 |
## License
|
52 |
+
MIT
|