close #4

add data class
moriaki3193 · Jan 26, 2019 · e1c2a58 · e1c2a58
1 parent e7a25da
commit e1c2a58
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/cdfm/consts.py b/cdfm/consts.py
@@ -1 +1,9 @@
 # -*- coding: utf-8 -*-
+"""Constants
+"""
+LABEL = 'label'
+QID = 'qid'
+EID = 'eid'
+CIDS = 'cids'
+FEATURES = 'features'
+PROXIMITIES = 'proximities'
diff --git a/cdfm/data.py b/cdfm/data.py
@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+"""Definition of records in dataset.
+"""
+from collections import namedtuple
+from .consts import LABEL, QID, EID, CIDS, FEATURES, PROXIMITIES
+
+
+Row = namedtuple('Row', (LABEL, QID, EID, CIDS, FEATURES, PROXIMITIES))
diff --git a/cdfm/types.py b/cdfm/types.py
@@ -1,8 +1,10 @@
 # -*- coding: utf-8 -*-
 """Type annotations.
 """
-from typing import Dict, Union
+from typing import Dict, List, Union
+from .data import Row
 
 
+Dataset = List[Row]
 EntityID = Union[int, str]
 EntIndMap = Dict[EntityID, int]