From d4dbc11c8b3dbbeea4ae901e66685edf6ff4bed9 Mon Sep 17 00:00:00 2001
From: kinghuin <kinghuin_chull@163.com>
Date: Tue, 31 Dec 2019 15:51:26 +0800
Subject: [PATCH] Update dataset.py (#295)

---
 paddlehub/dataset/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlehub/dataset/dataset.py b/paddlehub/dataset/dataset.py
index c7633cf7..a064cde3 100644
--- a/paddlehub/dataset/dataset.py
+++ b/paddlehub/dataset/dataset.py
@@ -167,7 +167,7 @@ class BaseDataset(object):
 
     def _load_label_data(self):
         with open(os.path.join(self.base_path, self.label_file), "r") as file:
-            return file.read().split("\n")
+            return file.read().strip().split("\n")
 
     def __str__(self):
         return "Dataset: %s with %i train examples, %i dev examples and %i test examples" % (
-- 
GitLab