|
a |
|
b/model/utils.py |
|
|
1 |
def age_vocab(max_age, mon=1, symbol=None): |
|
|
2 |
age2idx = {} |
|
|
3 |
idx2age = {} |
|
|
4 |
if symbol is None: |
|
|
5 |
symbol = ['PAD', 'UNK'] |
|
|
6 |
|
|
|
7 |
for i in range(len(symbol)): |
|
|
8 |
age2idx[str(symbol[i])] = i |
|
|
9 |
idx2age[i] = str(symbol[i]) |
|
|
10 |
|
|
|
11 |
if mon == 12: |
|
|
12 |
for i in range(max_age): |
|
|
13 |
age2idx[str(i)] = len(symbol) + i |
|
|
14 |
idx2age[len(symbol) + i] = str(i) |
|
|
15 |
elif mon == 1: |
|
|
16 |
for i in range(max_age * 12): |
|
|
17 |
age2idx[str(i)] = len(symbol) + i |
|
|
18 |
idx2age[len(symbol) + i] = str(i) |
|
|
19 |
else: |
|
|
20 |
age2idx = None |
|
|
21 |
idx2age = None |
|
|
22 |
return age2idx, idx2age |