Skip to content

Commit e94de3c

Browse files
authored
Bugfix: Select paswx paraphrases instead of non-paraphrases
1 parent 26b78d3 commit e94de3c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

data/pawsx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def __init__(self, language: str = "en", split: str = "validation"):
2323
self.split = split
2424
dataset = load_dataset("paws-x", language, split=self.split)
2525
# Only use sentence pairs that are paraphrases
26-
self.dataset = dataset.filter(lambda example: example["label"] == 0)
26+
self.dataset = dataset.filter(lambda example: example["label"] == 1)
2727

2828
def get_samples(self) -> List[PAWSXSample]:
2929
samples = []

0 commit comments

Comments
 (0)