Spaces:
Running
Running
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 |
|