diff options
| author | Christian Krinitsin <mail@krinitsin.com> | 2025-06-01 21:19:28 +0200 |
|---|---|---|
| committer | Christian Krinitsin <mail@krinitsin.com> | 2025-06-01 21:19:28 +0200 |
| commit | 05ed9b00104b3da2e9cb0d541b5c5bbceb027fde (patch) | |
| tree | d47bd9721b78db6131d56780e7230cf69412eaa6 /classification/output.py | |
| parent | cddf7dfa5a6e92a9057e7f6afae5d9b970585e6f (diff) | |
| download | emulator-bug-study-05ed9b00104b3da2e9cb0d541b5c5bbceb027fde.tar.gz emulator-bug-study-05ed9b00104b3da2e9cb0d541b5c5bbceb027fde.zip | |
adjust classifier
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: |