timestamp appended to output file name

This commit is contained in:
Andrea Pedrotti 2024-03-12 13:38:12 +01:00
parent 31fc225d0e
commit 650e764864
1 changed files with 5 additions and 1 deletions

View File

@ -1,6 +1,7 @@
import os
import pandas as pd
from pathlib import Path
from datetime import datetime
from dataManager.gFunDataset import SimpleGfunDataset
from gfun.generalizedFunnelling import GeneralizedFunnelling
@ -87,7 +88,10 @@ def save_inference_preds(preds, dataset_name, doc_ids, targets=None, category_ma
mapper = pd.read_csv(category_mapper).to_dict()["category"]
df["gfun_string_prediction"] = [mapper[p] for p in _preds]
output_file = f"{outdir}/{dataset_name}.csv"
timestamp = datetime.now()
formatted_timestamp = timestamp.strftime("%y%m%d_%H%M%S")
output_file = f"{outdir}/{dataset_name}_{formatted_timestamp}.csv"
print(f"Storing predicitons in: {output_file}")
df.to_csv(output_file, index=False)