diff options
Diffstat (limited to 'classification/output.py')
| -rw-r--r-- | classification/output.py | 12 |
1 files changed, 0 insertions, 12 deletions
diff --git a/classification/output.py b/classification/output.py deleted file mode 100644 index 971c3830..00000000 --- a/classification/output.py +++ /dev/null @@ -1,12 +0,0 @@ -from os import path, makedirs - -def output(text : str, category : str, labels : list, scores : list, identifier : str): - file_path = f"output/{category}/{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) |