Spaces:
Runtime error
Runtime error
Fix wrong sentiment mapping on the test dataset
Browse files- app/data.py +2 -2
app/data.py
CHANGED
|
@@ -333,8 +333,8 @@ def load_test(include_neutral: bool = False) -> tuple[list[str], list[int]]:
|
|
| 333 |
data["label"] = data["label"].map(
|
| 334 |
{
|
| 335 |
0: 0, # Negative
|
| 336 |
-
1:
|
| 337 |
-
2:
|
| 338 |
},
|
| 339 |
)
|
| 340 |
|
|
|
|
| 333 |
data["label"] = data["label"].map(
|
| 334 |
{
|
| 335 |
0: 0, # Negative
|
| 336 |
+
1: 2, # Neutral
|
| 337 |
+
2: 1, # Positive
|
| 338 |
},
|
| 339 |
)
|
| 340 |
|