summary refs log tree commit diff stats
path: root/classification
diff options
context:
space:
mode:
authorChristian Krinitsin <mail@krinitsin.com>2025-06-04 21:20:13 +0000
committerChristian Krinitsin <mail@krinitsin.com>2025-06-04 21:20:13 +0000
commit2773b5e236e2217a35a53dbc298757610c7bbbc6 (patch)
tree07b1cfc6ccd3a8f2808ffac94e8a4488079787a6 /classification
parent45cde517004371146f7533c59c8005b8b2629291 (diff)
downloadqemu-analysis-2773b5e236e2217a35a53dbc298757610c7bbbc6.tar.gz
qemu-analysis-2773b5e236e2217a35a53dbc298757610c7bbbc6.zip
add new iteration
Diffstat (limited to 'classification')
-rwxr-xr-xclassification/main.py5
1 files changed, 5 insertions, 0 deletions
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__":