From 5693e9e5809d31c7a5294dab5933d68714850bc5 Mon Sep 17 00:00:00 2001 From: Authur-Joker Date: Thu, 18 Nov 2021 15:45:11 +0800 Subject: [PATCH] dataloader altering --- dataloader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataloader.py b/dataloader.py index cff09ef..64f608d 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() -- GitLab