diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py index 011ec365a23d137f814e4dc261524629c659fd18..12280a99490424dbd5419ebd74a97a92cee5012d 100644 --- a/convlab/policy/emoTUS/evaluate.py +++ b/convlab/policy/emoTUS/evaluate.py @@ -236,8 +236,8 @@ class Evaluator: no_neutral=False) if self.use_sentiment: sent_score = sentiment_score( - gen_file['dialog']["golden_sentiment"], - gen_file['dialog']["gen_sentiment"], + self.r["golden_sentiment"], + self.r["gen_sentiment"], self.model_checkpoint, time=self.time)