mm/citybert
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

print

Browse Source
This commit is contained in:
mm 2023-05-04 22:00:59 +00:00
parent 502bbdba5e
commit 5f3174e6a3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
eval.py
View File

@ -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")