1 files changed, 5 insertions, 0 deletions
diff --git a/classification/main.py b/classification/main.py
index 93f92c8f..742bb92f 100755
--- a/classification/main.py
+++ b/classification/main.py
@@ -61,6 +61,11 @@ def main():
                 category = "other"
                 break
 
+        if all(i > 0.9 for i in result["scores"]):
+            category = "all"
+        elif all(i < 0.6 for i in result["scores"]):
+            category = "none"
+
         output(text, category, result['labels'], result['scores'], path.basename(bug))
 
 if __name__ == "__main__":