entities.py 文件源码

def __init__(self):
        self.entities = {
                '<cuisine>' : None,
                '<location>' : None,
                '<party_size>' : None,
                '<rest_type>' : None,
                }
        self.num_features = 4 # tracking 4 entities
        self.rating = None

        # constants
        self.party_sizes = ['??', '??', '??', '?', '??', '?', '??', '???', '??', '???', '??', '??']

        self.locations = ['??', '???', '???', '???', '??', '??', '??', '????', '??', '??', 'LA']

        self.cuisines = ['??','??','???', '????', '??', '??', '??', '??', '??', '???', '??', '???']

        self.rest_types = ['??', '?','??', '??']

        self.EntType = Enum('Entity Type', '<party_size> <location> <cuisine> <rest_type> <non_ent>')