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()
