def __init__(self, agents, args):
     # for now we only suppport dialog of 2 agents
     assert len(agents) == 2
     self.agents = agents
     self.args = args
     self.domain = domain.get_domain(args.domain)
     self.metrics = MetricsContainer()
     self._register_metrics()
Пример #2
0
 def __init__(self, agents, args, markable_detector,
              markable_detector_corpus):
     # For now we only suppport dialog of 2 agents
     assert len(agents) == 2
     self.agents = agents
     self.args = args
     self.domain = domain.get_domain(args.domain)
     self.metrics = MetricsContainer()
     self._register_metrics()
     self.markable_detector = markable_detector
     self.markable_detector_corpus = markable_detector_corpus
     self.selfplay_markables = {}
     self.selfplay_referents = {}