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}/*")):