diff options
| author | Christian Krinitsin <mail@krinitsin.com> | 2025-06-08 14:19:33 +0000 |
|---|---|---|
| committer | Christian Krinitsin <mail@krinitsin.com> | 2025-06-08 14:19:33 +0000 |
| commit | 9ebc3c7b58e0820054942a2e22b7c48889c3ee26 (patch) | |
| tree | 4aa48e6571ba9d616b61fde66be09aba9c9594f9 /classification | |
| parent | 3af0ee7c943b43fc12cca126c4cc03eef9f1191a (diff) | |
| download | qemu-analysis-9ebc3c7b58e0820054942a2e22b7c48889c3ee26.tar.gz qemu-analysis-9ebc3c7b58e0820054942a2e22b7c48889c3ee26.zip | |
add 012 result
Diffstat (limited to 'classification')
| -rwxr-xr-x | classification/classifier.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/classification/classifier.py b/classification/classifier.py index d29d385a2..5c81f1818 100755 --- a/classification/classifier.py +++ b/classification/classifier.py @@ -11,8 +11,8 @@ parser.add_argument('--model', default="facebook/bart-large-mnli", type=str, hel parser.add_argument('--compare', nargs='?', const="MoritzLaurer/deberta-v3-large-zeroshot-v2.0", type=str, help="second model for comparison") args = parser.parse_args() -positive_categories = ['semantic'] -negative_categories = ['other', 'boot', 'network', 'KVM', 'vnc', 'graphic', 'device', 'socket', 'debug', 'files', 'PID', 'permissions', 'performance'] +positive_categories = ['semantic', 'TCG', 'assembly', 'architecture', 'mistranslation', 'register', 'x86', 'arm', 'risc-v'] +negative_categories = ['other', 'boot', 'network', 'kernel virtual machine', 'vnc', 'graphic', 'device', 'socket', 'debug', 'files', 'PID', 'permissions', 'performance'] categories = positive_categories + negative_categories def list_files_recursive(directory): |