Skip to content
Snippets Groups Projects
Select Git revision
  • 6a800c444d6aaff27aaa24cc8a090ad1333beb27
  • main default protected
  • gentus-public
3 results

rule_agent.py

Blame
    • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    rule_agent.py 762 B
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29 
    

    

    

    '''
rule_agent.py - An rule-based dialogue agent class
==========================================================================

Build up an pipeline agent with nlu, dst, policy and nlg.

@author: linh
'''


from convlab2.dialog_agent.agent import DialogueAgent
from convlab2.nlu.jointBERT.multiwoz import BERTNLU
from convlab2.dst.rule.multiwoz import RuleDST
from convlab2.policy.rule.multiwoz import RulePolicy
from convlab2.nlg.template.multiwoz import TemplateNLG


class Agent(DialogueAgent):

    def __init__(self):
        nlu = BERTNLU()
        dst = RuleDST()
        policy = RulePolicy()

        nlg = TemplateNLG(is_user=False)
        super().__init__(nlu, dst, policy, nlg)

        self.agent_name = "BERTNLU-RuleDST-RulePolicy-TemplateNLG"