Browse Source

给 dataset 增加了get_field_names

tags/v0.4.10
ChenXin 6 years ago
parent
commit
3465d2a08f
1 changed files with 8 additions and 0 deletions
  1. +8
    -0
      fastNLP/core/dataset.py

+ 8
- 0
fastNLP/core/dataset.py View File

@@ -501,6 +501,14 @@ class DataSet(object):
""" """
return self.field_arrays return self.field_arrays



def get_field_names(self)->list:
"""返回一个list,包含所有 field 的名字

:return: list:
"""
return sorted(self.field_arrays.keys())
def get_length(self): def get_length(self):
"""获取DataSet的元素数量 """获取DataSet的元素数量




Loading…
Cancel
Save