Skip to content
Snippets Groups Projects
Commit 8726a586 authored by Hsien-Chin Lin's avatar Hsien-Chin Lin
Browse files

wip

parent 323b6622
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
convlab/policy/emoTUS/emoTUS.py
+ 1
− 1
View file @ 8726a586
...@@ -121,7 +121,7 @@ class UserActionPolicy(GenTUSUserActionPolicy): ...@@ -121,7 +121,7 @@ class UserActionPolicy(GenTUSUserActionPolicy):
print("-"*20) print("-"*20)
return action return action
def _generate_action(self, raw_inputs, mode="max", allow_general_intent=True, emotion_mode="max"): def _generate_action(self, raw_inputs, mode="max", allow_general_intent=True, emotion_mode="normal"):
self.kg.parse_input(raw_inputs) self.kg.parse_input(raw_inputs)
model_input = self.vector.encode(raw_inputs, self.max_in_len) model_input = self.vector.encode(raw_inputs, self.max_in_len)
# start token # start token
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment