diff --git a/infer.py b/infer.py index 58075cc..004eaa4 100644 --- a/infer.py +++ b/infer.py @@ -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)