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