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

full training process on US cities

main
mm 2 years ago
parent
282c0466d8
commit
f193018ac2
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      eval.py

2
eval.py
View File

@ -66,7 +66,7 @@ if __name__ == "__main__":
model_name = "sentence-transformers/all-MiniLM-L6-v2" model_name = "sentence-transformers/all-MiniLM-L6-v2"
base_model = SentenceTransformer(model_name, device="cuda") 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) # data_sample = data.sample(1_000)
checkpoint_dir = "checkpoints_absmax_split" # no slash 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}/*")):

Write Preview
Loading…
Cancel
Save