|
|
|
@ -69,7 +69,8 @@ if __name__ == "__main__": |
|
|
|
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}/*")): |
|
|
|
for checkpoint in sorted(glob.glob(f"{checkpoint_dir}/*"))[14::]: |
|
|
|
print(f"Evaluating {checkpoint}") |
|
|
|
data_sample = data.sample(1_000) |
|
|
|
trained_model = SentenceTransformer(checkpoint, device="cuda") |
|
|
|
|
|
|
|
|
