[edit] update plot position
Browse files
README.md
CHANGED
@@ -11,9 +11,11 @@ LLM trained to predict a character's emotional response in the given situation
|
|
11 |
* Trained to predict in a structured output format.
|
12 |
|
13 |
Prediction Performance:
|
14 |
-
|
|
15 |
-
|
16 |
-
| <img src="./assets/qwen2_5-3b-baseline_perf.png" alt="baseline_perf" width="
|
|
|
|
|
17 |
|
18 |
## Quickstart
|
19 |
The model is trained to predict in the following schema
|
|
|
11 |
* Trained to predict in a structured output format.
|
12 |
|
13 |
Prediction Performance:
|
14 |
+
| Setting | Performance by Emotion|
|
15 |
+
| --- | --- |
|
16 |
+
| Pretrained | <img src="./assets/qwen2_5-3b-baseline_perf.png" alt="baseline_perf" width="100%" /> |
|
17 |
+
| Tuned | <img src="./assets/finetuned_perf.png" alt="trained_perf" width="100%" /> |
|
18 |
+
|
19 |
|
20 |
## Quickstart
|
21 |
The model is trained to predict in the following schema
|