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