def __init__(self, cfg): super(TemplateNLG, self).__init__(cfg) # load templates if 'model' in self.cfg['NLG']['Template']: self.load_templates(self.cfg['NLG']['Template']['model']) # load ontology self.ontology = Ontology() if 'ontology' in self.cfg['NLG']['Template']: self.ontology.load(cfg['NLG']['Template']['ontology']) # initialize pre- and post-processing self.preprocessing = None self.postprocessing = None if 'preprocessing_cls' in self.cfg['NLG']['Template']: if 'preprocessing_config' in self.cfg['NLG']['Template']: self.preprocessing = self.cfg['NLG']['Template'][ 'preprocessing_cls']( self.ontology, self.cfg['NLG']['Template']['preprocessing_config']) else: self.preprocessing = self.cfg['NLG']['Template'][ 'preprocessing_cls'](self.ontology) if 'postprocessing_cls' in self.cfg['NLG']['Template']: self.postprocessing = self.cfg['NLG']['Template'][ 'postprocessing_cls']()
def __init__(self, cfg): self.cfg = cfg self.ontology = Ontology(self.cfg['DM']['ontology']) self.dialogue_state_class = self.cfg['DM']['dialogue_state']['type'] self.dialogue_policy_class = self.cfg['DM']['dialogue_policy']['type'] self.last_system_dialogue_act = None self.new_dialogue()
def setUp(self): super(TestPTICSHDCPolicy, self).setUp() self.cfg = self._get_cfg() self.ontology = Ontology(self.cfg['DM']['ontology']) self.mox = mox.Mox()