Diff of /model/utils.py [000000] .. [bad60c]

Switch to unified view

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