diff options
| author | Christian Krinitsin <mail@krinitsin.com> | 2025-06-04 20:55:58 +0000 |
|---|---|---|
| committer | Christian Krinitsin <mail@krinitsin.com> | 2025-06-04 20:55:58 +0000 |
| commit | 45cde517004371146f7533c59c8005b8b2629291 (patch) | |
| tree | 94d550cdbb053c271101d9cc07e5dddc0a0c7edb /classification/main.py | |
| parent | 0ffbcc8b5f9f084a14fa6e4b93095758b63c465d (diff) | |
| download | qemu-analysis-45cde517004371146f7533c59c8005b8b2629291.tar.gz qemu-analysis-45cde517004371146f7533c59c8005b8b2629291.zip | |
add new results
Diffstat (limited to 'classification/main.py')
| -rwxr-xr-x | classification/main.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/classification/main.py b/classification/main.py index 95ef80985..93f92c8fc 100755 --- a/classification/main.py +++ b/classification/main.py @@ -8,8 +8,8 @@ parser = ArgumentParser(prog='main.py') parser.add_argument('-m', '--minimal', action='store_true') args = parser.parse_args() -positive_categories = ['semantic', 'mistranslation', 'instruction', 'assembly'] # to add: register -negative_categories = ['other', 'boot', 'network', 'KVM', 'vnc', 'graphic', 'device', 'socket'] # to add: performance +positive_categories = ['semantic'] +negative_categories = ['other', 'boot', 'network', 'KVM', 'vnc', 'graphic', 'device', 'socket', 'debug', 'files', 'PID', 'permissions', 'performance'] categories = positive_categories + negative_categories def list_files_recursive(directory): @@ -38,7 +38,7 @@ def main(): bugs = list_files_recursive("../results/scraper/mailinglist") if args.minimal: - bugs = bugs + list_files_recursive("../results/gitlab/scraper/semantic_issues") + bugs = bugs + list_files_recursive("../results/scraper/gitlab/semantic_issues") else: bugs = bugs + list_files_recursive("../results/scraper/launchpad") bugs = bugs + list_files_recursive("../results/scraper/gitlab/issues_text") @@ -57,6 +57,10 @@ def main(): category = label break + if label == "semantic" and score <= 0.91: + category = "other" + break + output(text, category, result['labels'], result['scores'], path.basename(bug)) if __name__ == "__main__": |