Skip to content

Commit cc5453e

Browse files
committed
good hyper parameters for the task1 spot dataset
1 parent b4bc774 commit cc5453e

6 files changed

Lines changed: 16 additions & 16 deletions

File tree

python/benchmark_task1_large.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def _fmt(val: float | None) -> str:
5454
def _fmt_recall(val: float | None) -> str:
5555
if val is None:
5656
return "—"
57-
return f"{val:.2f}%"
57+
return f"{val * 100.0:.2f}%"
5858

5959

6060
def _strip_ansi(text: str) -> str:

python/benchmark_task1_small.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def _fmt(val: float | None) -> str:
5555
def _fmt_recall(val: float | None) -> str:
5656
if val is None:
5757
return "—"
58-
return f"{val:.2f}%"
58+
return f"{val * 100.0:.2f}%"
5959

6060

6161
def _strip_ansi(text: str) -> str:

python/benchmark_task1_spot.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -36,13 +36,13 @@ class ModeConfig:
3636

3737

3838
MODES: list[ModeConfig] = [
39-
ModeConfig(name="mode1", mode="fp16", label="FP16 Build+Explore", settings="M=24, MaxDist=100", max_dist=100),
40-
# ModeConfig(name="mode2", mode="evp-linear", label="EVP linear search", settings="—"),
41-
ModeConfig(name="mode3", mode="evp", label="EVP Build+Explore", settings="M=24, MaxDist=200"),
42-
ModeConfig(name="mode4", mode="evp-rerank", label="EVP Build+Explore+Rerank", settings="M=24, MaxDist=200, evpK=50", evp_k=50),
43-
ModeConfig(name="mode5", mode="evp-build-fp16-external-search", label="EVP build+FP16 Explore", settings="M=24, MaxDist=200"),
44-
ModeConfig(name="mode6", mode="evp-asymmetric", label="EVP build+Asym Explore", settings="M=24, MaxDist=200"),
45-
ModeConfig(name="mode7", mode="evp-asymmetric-rerank", label="EVP build+Asym+Rerank", settings="M=24, MaxDist=200, evpK=50", evp_k=50),
39+
ModeConfig(name="mode1", mode="fp16", label="FP16 Build+Explore", settings="M=16, MaxDist=100", max_dist=100),
40+
ModeConfig(name="mode2", mode="evp-linear", label="EVP linear search", settings="—"),
41+
ModeConfig(name="mode3", mode="evp", label="EVP Build+Explore", settings="M=16, MaxDist=200"),
42+
ModeConfig(name="mode4", mode="evp-rerank", label="EVP Build+Explore+Rerank", settings="M=16, MaxDist=100, evpK=50", max_dist=100, evp_k=50),
43+
ModeConfig(name="mode5", mode="evp-build-fp16-external-search", label="EVP build+FP16 Explore", settings="M=16, MaxDist=200"),
44+
ModeConfig(name="mode6", mode="evp-asymmetric", label="EVP build+Asym Explore", settings="M=16, MaxDist=200"),
45+
ModeConfig(name="mode7", mode="evp-asymmetric-rerank", label="EVP build+Asym+Rerank", settings="M=16, MaxDist=200, evpK=50", evp_k=50),
4646
]
4747

4848

@@ -55,7 +55,7 @@ def _fmt(val: float | None) -> str:
5555
def _fmt_recall(val: float | None) -> str:
5656
if val is None:
5757
return "—"
58-
return f"{val:.2f}%"
58+
return f"{val * 100.0:.2f}%"
5959

6060

6161
def _strip_ansi(text: str) -> str:
@@ -73,9 +73,9 @@ def run_mode(runner: Task1Runner, cfg: ModeConfig, num_threads: int) -> Task1Res
7373
size="spot",
7474
max_dist=cfg.max_dist,
7575
threads=num_threads,
76-
non_zeros=300,
77-
k_graph=24,
78-
k_ext=24,
76+
non_zeros=200,
77+
k_graph=16,
78+
k_ext=16,
7979
prune_worst=8,
8080
)
8181
if cfg.evp_k is not None:

python/submission_task1_large.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def _fmt(val: float | None) -> str:
152152
def _fmt_recall(val: float | None) -> str:
153153
if val is None:
154154
return "—"
155-
return f"{val:.2f}%"
155+
return f"{val * 100.0:.2f}%"
156156

157157

158158
def run_mode(runner: Task1Runner, cfg: ModeConfig, num_threads: int) -> Task1Result | None:

python/submission_task1_small.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def _fmt(val: float | None) -> str:
152152
def _fmt_recall(val: float | None) -> str:
153153
if val is None:
154154
return "—"
155-
return f"{val:.2f}%"
155+
return f"{val * 100.0:.2f}%"
156156

157157

158158
def run_mode(runner: Task1Runner, cfg: ModeConfig, num_threads: int) -> Task1Result | None:

submission/search.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@
7878
],
7979
# 384-dim gooaq spot-check (different family; smoke test only, non_zeros<dim)
8080
"gooaq-small": [
81-
dict(mode="mode4", non_zeros=300, k_graph=24, k_ext=24, eps_ext=0.001, prune_worst=8, evpK=50, max_dist=[200, 400, 800]),
81+
dict(mode="mode4", non_zeros=200, k_graph=16, k_ext=16, eps_ext=0.001, prune_worst=8, evpK=50, max_dist=[100, 150, 200]),
8282
],
8383
}
8484

0 commit comments

Comments
 (0)