From 2773b5e236e2217a35a53dbc298757610c7bbbc6 Mon Sep 17 00:00:00 2001 From: Christian Krinitsin Date: Wed, 4 Jun 2025 21:20:13 +0000 Subject: add new iteration --- classification/main.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'classification/main.py') diff --git a/classification/main.py b/classification/main.py index 93f92c8fc..742bb92f3 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__": -- cgit 1.4.1