diff --git a/dataloader.py b/dataloader.py index cff09ef77c34dea2154700ccda6a76f9eb2761b6..64f608dab4d8c47f07a492bc510f3b7947460ef9 100644 --- a/dataloader.py +++ b/dataloader.py @@ -79,7 +79,7 @@ class NERDataset(Dataset): def read_file(self): - with open(self.config.test_file, "r", encoding="utf-8") as f: + with open(self.file_path, "r", encoding="utf-8") as f: lines, words, labels = [], [], [] for line in f.readlines(): contends = line.strip()