Tymec commited on
Commit
3a96048
1 Parent(s): 53bc5fb

Fix wrong sentiment mapping on the test dataset

Browse files
Files changed (1) hide show
  1. 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: 1, # Neutral
337
- 2: 2, # Positive
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