Update README.md
Browse files
README.md
CHANGED
@@ -5,9 +5,9 @@ license: cc-by-sa-4.0
|
|
5 |
This repository contains three binomial regression models designed to predict VAD (Valence, Arousal, Dominance) scores for text inputs.
|
6 |
Each model is stored as a separate pickle (.pkl) file:
|
7 |
|
8 |
-
valence_model.pkl
|
9 |
-
arousal_model.pkl
|
10 |
-
dominance_model.pkl
|
11 |
All scores are normalized on a scale from 0 to 1.
|
12 |
|
13 |
Before making predictions, input text must be converted into embeddings using the [Knesset-multi-e5-large](https://huggingface.co/GiliGold/Knesset-multi-e5-large) model. The embeddings are then fed into the regression models.
|
|
|
5 |
This repository contains three binomial regression models designed to predict VAD (Valence, Arousal, Dominance) scores for text inputs.
|
6 |
Each model is stored as a separate pickle (.pkl) file:
|
7 |
|
8 |
+
- **valence_model.pkl**: Predicts the Valence score (positivity/negativity).
|
9 |
+
- **arousal_model.pkl**: Predicts the Arousal score (level of excitement or calm).
|
10 |
+
- **dominance_model.pkl**: Predicts the Dominance score (sense of control or influence).
|
11 |
All scores are normalized on a scale from 0 to 1.
|
12 |
|
13 |
Before making predictions, input text must be converted into embeddings using the [Knesset-multi-e5-large](https://huggingface.co/GiliGold/Knesset-multi-e5-large) model. The embeddings are then fed into the regression models.
|