Add default value of valid_labels

action_generation.py

@@ -56,8 +56,21 @@ Examples:
 # TODO: Define external resources urls if needed
 BAD_WORDS_URL = "http://url/to/external/resource/bad_words.txt"
 
+VALID_LABELS = [
+    "/開箱",
+    "/教學",
+    "/表達",
+    "/分享/外部資訊",
+    "/分享/個人資訊",
+    "/推薦/產品",
+    "/推薦/服務",
+    "/推薦/其他",
+    ""
+]
+
 class BaseEvaluater:
     eps = 1e-8
+    valid_labels = None
 
     def __call__(self, preds, labels):
         return self._compute(preds, labels)
@@ -78,10 +91,15 @@ class BaseEvaluater:
             "recall": round(recall, 4),
             "f1": round(f1, 4)
         }
+    
+    def _init_valid_labels(self):
+        if self.valid_labels is None:
+            self.valid_labels = VALID_LABELS
 
 class ClassEvaluater(BaseEvaluater):
     def __init__(self, valid_labels=None):
         self.valid_labels = valid_labels
+        self._init_valid_labels()
 
     def __call__(self, preds, labels):
         preds = map(self.extract_class, preds)
@@ -92,7 +110,6 @@ class ClassEvaluater(BaseEvaluater):
         return self._compute(preds, labels)
 
     def extract_valid(self, tags):
-        # TODO: if valid_labels is None:
         tags = list(filter(lambda tag: tag in self.valid_labels, tags))
         return tags
 
@@ -123,6 +140,7 @@ class ClassEvaluater(BaseEvaluater):
 class PhraseEvaluater(BaseEvaluater):
     def __init__(self, valid_labels=None):
         self.valid_labels = valid_labels
+        self._init_valid_labels()
 
     def __call__(self, preds, labels):
         preds = map(self.extract_phrase, preds)