-
Notifications
You must be signed in to change notification settings - Fork 0
/
featuremap.py
executable file
·41 lines (36 loc) · 1.2 KB
/
featuremap.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#!/usr/bin/python
def feature_to_str(feature):
if isinstance(feature, tuple):
return '%s %s' % (feature[0] , feature[1])
return feature
def load_feature_map(filename='features.idx'):
feature_map_toidx = {}
feature_map_fromidx = []
fin = open(filename)
for n, line in enumerate(fin):
line = line.strip()
section, delim, feature = line.partition(' ')
feature_map_toidx[ (section, feature) ] = n
feature_map_fromidx.append( (section, feature) )
fin.close()
return (feature_map_toidx, feature_map_fromidx)
def load_concept_map(filename='concepts.idx'):
concept_map_toidx = {}
concept_map_fromidx = []
fin = open(filename)
for n, line in enumerate(fin):
line = line.strip()
concept_map_toidx[line] = n
concept_map_fromidx.append(line)
fin.close()
return (concept_map_toidx, concept_map_fromidx)
def load_word_map(filename='words.idx'):
word_map_toidx = {}
word_map_fromidx = []
fin = open(filename)
for n, line in enumerate(fin):
line = line.strip()
word_map_toidx[line] = n
word_map_fromidx.append(line)
fin.close()
return (word_map_toidx, word_map_fromidx)