classifier: iterates through mailing list

1 files changed, 12 insertions, 0 deletions

diff --git a/classification/output.py b/classification/output.py
new file mode 100644
index 000000000..df64dbccd
--- /dev/null
+++ b/classification/output.py
@@ -0,0 +1,12 @@
+from os import path, makedirs
+
+def output(text : str, labels : list, scores : list, identifier : str):
+    file_path = f"output/{labels[0]}/{identifier}"
+    makedirs(path.dirname(file_path), exist_ok = True)
+
+    with open(file_path, "w") as file:
+        for label, score in zip(labels, scores):
+            file.write(f"{label}: {score:.3f}\n")
+
+        file.write("\n")
+        file.write(text)