summary refs log tree commit diff stats
path: root/classification/output.py
blob: df64dbccd2c9982bedb00e243f27ab307f8de6e0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
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)