From f193018ac24ec31eb1e58b22418abc35f595cc00 Mon Sep 17 00:00:00 2001 From: mm Date: Thu, 4 May 2023 19:05:46 +0000 Subject: [PATCH] full training process on US cities --- eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eval.py b/eval.py index 7874dae..70ad0e2 100644 --- a/eval.py +++ b/eval.py @@ -66,7 +66,7 @@ if __name__ == "__main__": model_name = "sentence-transformers/all-MiniLM-L6-v2" base_model = SentenceTransformer(model_name, device="cuda") - data = pd.read_csv("city_distances_sample.csv") + data = pd.read_csv("city_distances_full.csv") # data_sample = data.sample(1_000) checkpoint_dir = "checkpoints_absmax_split" # no slash for checkpoint in sorted(glob.glob(f"{checkpoint_dir}/*")):