summary refs log tree commit diff stats
path: root/classification/test.py
diff options
context:
space:
mode:
Diffstat (limited to 'classification/test.py')
-rwxr-xr-xclassification/test.py18
1 files changed, 0 insertions, 18 deletions
diff --git a/classification/test.py b/classification/test.py
deleted file mode 100755
index bcd6b439..00000000
--- 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("")