diff options
| author | Christian Krinitsin <mail@krinitsin.com> | 2025-06-01 21:35:14 +0200 |
|---|---|---|
| committer | Christian Krinitsin <mail@krinitsin.com> | 2025-06-01 21:35:14 +0200 |
| commit | 3e4c5a6261770bced301b5e74233e7866166ea5b (patch) | |
| tree | 9379fddaba693ef8a045da06efee8529baa5f6f4 /classification/test.py | |
| parent | e5634e2806195bee44407853c4bf8776f7abfa4f (diff) | |
| download | qemu-analysis-3e4c5a6261770bced301b5e74233e7866166ea5b.tar.gz qemu-analysis-3e4c5a6261770bced301b5e74233e7866166ea5b.zip | |
clean up repository
Diffstat (limited to 'classification/test.py')
| -rwxr-xr-x | classification/test.py | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/classification/test.py b/classification/test.py deleted file mode 100755 index bcd6b4392..000000000 --- a/classification/test.py +++ /dev/null @@ -1,18 +0,0 @@ -from os import listdir, path - -directory : str = "./test_input" - -def test(classifier, categories): - for name in listdir(directory): - if name == "README.md": - continue - - with open(path.join(directory, name), "r") as file: - text = file.read() - - result = classifier(text, categories, multi_label=True) - - print(name) - for label, score in zip(result["labels"], result["scores"]): - print(f"{label}: {score:.3f}") - print("") |