lstm_model.py 文件源码

def __init__(self, input_size, num_hidden, num_classes, utter_embed, bow_utter_embed, config):
        self.input_size = input_size
        self.num_hidden = num_hidden
        self.num_classes = num_classes
        self.utter_embed = utter_embed
        self.bow_utter_embed = bow_utter_embed
        self.logger = config.logger
        self.config = config

        self.cate_mapping_dict = joblib.load('./dbdc3/data/cate_mapping_dict')