[#42] update stance labels in postprocess

aisingapore · Feb 15, 2022 · 9f8f730 · 9f8f730
1 parent b330dc0
commit 9f8f730
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/demo_api/dual_bert/usage.py b/demo_api/dual_bert/usage.py
@@ -17,11 +17,6 @@
 model.eval()
 
 examples = [
-    [
-        "#4U9525: Robin names Andreas Lubitz as the copilot in the flight deck who crashed the aircraft.",
-        "@thatjohn @mschenk",
-        "@thatjohn Have they named the pilot?",
-    ],
     [
         "#4U9525: Robin names Andreas Lubitz as the copilot in the flight deck who crashed the aircraft.",
         "@thatjohn @mschenk",

diff --git a/sgnlp/models/dual_bert/postprocess.py b/sgnlp/models/dual_bert/postprocess.py
@@ -27,7 +27,7 @@ def __call__(self, model_outputs: [DualBertModelOutput], stance_label_mask):
                     temp_2.append(self.stance_labels[stance_label_idx[i][j]])
                 else:
                     break
-            stance_labels.append(temp_2)
+            stance_labels.append(temp_2[1:]) #first post should not have a stance label
 
         return {
             "rumor_labels": rumor_labels,