diff options
| author | Christian Krinitsin <mail@krinitsin.com> | 2025-06-04 21:20:13 +0000 |
|---|---|---|
| committer | Christian Krinitsin <mail@krinitsin.com> | 2025-06-04 21:20:13 +0000 |
| commit | 2773b5e236e2217a35a53dbc298757610c7bbbc6 (patch) | |
| tree | 07b1cfc6ccd3a8f2808ffac94e8a4488079787a6 /classification/main.py | |
| parent | 45cde517004371146f7533c59c8005b8b2629291 (diff) | |
| download | emulator-bug-study-2773b5e236e2217a35a53dbc298757610c7bbbc6.tar.gz emulator-bug-study-2773b5e236e2217a35a53dbc298757610c7bbbc6.zip | |
add new iteration
Diffstat (limited to '')
| -rwxr-xr-x | classification/main.py | 5 |
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__": |