From 117f556a10c81c7a5b16e4826976145d61d700e3 Mon Sep 17 00:00:00 2001 From: yh Date: Tue, 21 May 2019 18:03:50 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BA=86=E4=B8=80=E4=B8=AAty?= =?UTF-8?q?po?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/io/dataset_loader.py | 16 ++++++++-------- legacy/api/api.py | 2 +- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/fastNLP/io/dataset_loader.py b/fastNLP/io/dataset_loader.py index b820af44..8273d2f8 100644 --- a/fastNLP/io/dataset_loader.py +++ b/fastNLP/io/dataset_loader.py @@ -214,27 +214,27 @@ class ConllLoader(DataSetLoader): 11:N Predicate Arguments N Coreference - :param headers: 每一列数据的名称,需为List or Tuple of str。``header`` 与 ``indexs`` 一一对应 - :param indexs: 需要保留的数据列下标,从0开始。若为 ``None`` ,则所有列都保留。Default: ``None`` + :param headers: 每一列数据的名称,需为List or Tuple of str。``header`` 与 ``indexes`` 一一对应 + :param indexes: 需要保留的数据列下标,从0开始。若为 ``None`` ,则所有列都保留。Default: ``None`` :param dropna: 是否忽略非法数据,若 ``False`` ,遇到非法数据时抛出 ``ValueError`` 。Default: ``False`` """ - def __init__(self, headers, indexs=None, dropna=False): + def __init__(self, headers, indexes=None, dropna=False): super(ConllLoader, self).__init__() if not isinstance(headers, (list, tuple)): raise TypeError('invalid headers: {}, should be list of strings'.format(headers)) self.headers = headers self.dropna = dropna - if indexs is None: - self.indexs = list(range(len(self.headers))) + if indexes is None: + self.indexes = list(range(len(self.headers))) else: - if len(indexs) != len(headers): + if len(indexes) != len(headers): raise ValueError - self.indexs = indexs + self.indexes = indexes def load(self, path): ds = DataSet() - for idx, data in _read_conll(path, indexes=self.indexs, dropna=self.dropna): + for idx, data in _read_conll(path, indexes=self.indexes, dropna=self.dropna): ins = {h: data[i] for i, h in enumerate(self.headers)} ds.append(Instance(**ins)) return ds diff --git a/legacy/api/api.py b/legacy/api/api.py index 0b5d3cd3..d5d1df6b 100644 --- a/legacy/api/api.py +++ b/legacy/api/api.py @@ -101,7 +101,7 @@ class ConllxDataLoader(ConllLoader): indexs = [ 1, 3, 6, 7, ] - super(ConllxDataLoader, self).__init__(headers=headers, indexs=indexs) + super(ConllxDataLoader, self).__init__(headers=headers, indexes=indexs) class API: