Skip to content
Snippets Groups Projects
Commit 6a8950fa authored by zqwerty's avatar zqwerty Committed by zhuqi
Browse files

change initiative from 4 to randint(2,4)

parent 1921dfd5
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
convlab2/policy/rule/multiwoz/policy_agenda_multiwoz.py
+ 2
− 2
View file @ 6a8950fa
...@@ -111,8 +111,8 @@ class UserPolicyAgendaMultiWoz(Policy): ...@@ -111,8 +111,8 @@ class UserPolicyAgendaMultiWoz(Policy):
self.agenda.close_session() self.agenda.close_session()
# A -> A' + user_action # A -> A' + user_action
# action = self.agenda.get_action(random.randint(1, self.max_initiative)) action = self.agenda.get_action(random.randint(2, self.max_initiative))
action = self.agenda.get_action(self.max_initiative) # action = self.agenda.get_action(self.max_initiative)
# transform to DA # transform to DA
action = self._transform_usract_out(action) action = self._transform_usract_out(action)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment