Skip to content

Commit 8e5a560

Browse files
JeadieJeadie
committed
rewrite algos.yml to ann_benchmarks/algorithms/{module_name}/config.yml
1 parent 8a8659e commit 8e5a560

File tree

46 files changed

+1875
-8
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

46 files changed

+1875
-8
lines changed

‎algos.yaml‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ float:
360360
ball:
361361
args: *treeargs
362362
ckdtree:
363-
disable: true
363+
disabled: true
364364
docker-tag: ann-benchmarks-scipy
365365
module: ann_benchmarks.algorithms.ckdtree
366366
constructor: CKDTree
@@ -719,7 +719,7 @@ float:
719719
args: [[100], [.nan], [4], ["squared_l2"]]
720720
query-args: [[[2, 20], [3, 20], [3, 30], [4, 30], [5, 40], [8, 80]]]
721721
kgraph:
722-
disable: true
722+
disabled: true
723723
docker-tag: ann-benchmarks-kgraph
724724
module: ann_benchmarks.algorithms.kgraph
725725
constructor: KGraph
@@ -1008,7 +1008,7 @@ float:
10081008
args : [{'l_build': 125, 'max_outdegree': 32, 'alpha': 1, 'chunks': 42}]
10091009
query-args : [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
10101010
puffinn:
1011-
disable: true
1011+
disabled: true
10121012
docker-tag: ann-benchmarks-puffinn
10131013
module: ann_benchmarks.algorithms.puffinn
10141014
constructor: Puffinn
@@ -1021,7 +1021,7 @@ float:
10211021
]
10221022
query-args: [[0.1, 0.2, 0.5, 0.7, 0.9, 0.95, 0.99]]
10231023
kgraph:
1024-
disable: true
1024+
disabled: true
10251025
docker-tag: ann-benchmarks-kgraph
10261026
module: ann_benchmarks.algorithms.kgraph
10271027
constructor: KGraph
@@ -1081,7 +1081,7 @@ float:
10811081
- False
10821082
query-args: [[120, 80, 60, 40, 20, 10, 8, 4, 2]]
10831083
rpforest:
1084-
disable: true
1084+
disabled: true
10851085
docker-tag: ann-benchmarks-rpforest
10861086
module: ann_benchmarks.algorithms.rpforest
10871087
constructor: RPForest
@@ -1258,7 +1258,7 @@ bit:
12581258
args: { "d" : "@dimension", "r" : [0, 2, 3],
12591259
"chunk-factor" : [0.2, 0.5, 0.8] }
12601260
kgraph:
1261-
disable: true
1261+
disabled: true
12621262
docker-tag: ann-benchmarks-kgraph
12631263
module: ann_benchmarks.algorithms.kgraph
12641264
constructor: KGraph
@@ -1449,7 +1449,7 @@ bit:
14491449
- False
14501450
query-args: [[120, 80, 60, 40, 20, 10, 8, 4, 2]]
14511451
BallTree(nmslib):
1452-
disable: true
1452+
disabled: true
14531453
docker-tag: ann-benchmarks-nmslib
14541454
module: ann_benchmarks.algorithms.nmslib
14551455
constructor: NmslibReuseIndex

‎ann_benchmarks/algorithms/annoy/config.yml‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
bit:
2+
hamming:
3+
- base_args: ['@metric']
4+
constructor: Annoy
5+
disabled: false
6+
docker_tag: ann-benchmarks-annoy
7+
module: ann_benchmarks.algorithms.annoy
8+
name: annoy
9+
run_groups:
10+
annoy:
11+
arg_groups: []
12+
args: [[100, 200, 400]]
13+
query_args: [[100, 200, 400, 1000, 2000, 4000, 10000, 20000, 40000, 100000,
14+
200000, 400000]]
15+
float:
16+
any:
17+
- base_args: ['@metric']
18+
constructor: Annoy
19+
disabled: false
20+
docker_tag: ann-benchmarks-annoy
21+
module: ann_benchmarks.algorithms.annoy
22+
name: annoy
23+
run_groups:
24+
annoy:
25+
arg_groups: []
26+
args: [[100, 200, 400]]
27+
query_args: [[100, 200, 400, 1000, 2000, 4000, 10000, 20000, 40000, 100000,
28+
200000, 400000]]

‎ann_benchmarks/algorithms/balltree/config.yml‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
float:
2+
any:
3+
- base_args: ['@metric']
4+
constructor: BallTree
5+
disabled: true
6+
docker_tag: ann-benchmarks-sklearn
7+
module: ann_benchmarks.algorithms.balltree
8+
name: ball
9+
run_groups:
10+
ball:
11+
arg_groups: []
12+
args: [[10, 20, 40, 100, 200, 400, 1000]]
13+
query_args: []

‎ann_benchmarks/algorithms/bruteforce/config.yml‎

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
bit:
2+
jaccard:
3+
- base_args: ['@metric']
4+
constructor: BruteForceBLAS
5+
disabled: false
6+
docker_tag: ann-benchmarks-sklearn
7+
module: ann_benchmarks.algorithms.bruteforce
8+
name: bf
9+
run_groups:
10+
base:
11+
arg_groups: []
12+
args: {}
13+
query_args: []
14+
float:
15+
any:
16+
- base_args: ['@metric']
17+
constructor: BruteForce
18+
disabled: true
19+
docker_tag: ann-benchmarks-sklearn
20+
module: ann_benchmarks.algorithms.bruteforce
21+
name: bruteforce
22+
run_groups:
23+
empty:
24+
arg_groups: []
25+
args: []
26+
query_args: []
27+
- base_args: ['@metric']
28+
constructor: BruteForceBLAS
29+
disabled: false
30+
docker_tag: ann-benchmarks-sklearn
31+
module: ann_benchmarks.algorithms.bruteforce
32+
name: bruteforce-blas
33+
run_groups:
34+
empty:
35+
arg_groups: []
36+
args: []
37+
query_args: []

‎ann_benchmarks/algorithms/ckdtree/config.yml‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
float:
2+
any:
3+
- base_args: ['@metric']
4+
constructor: CKDTree
5+
disabled: true
6+
docker_tag: ann-benchmarks-scipy
7+
module: ann_benchmarks.algorithms.ckdtree
8+
name: ckdtree
9+
run_groups:
10+
ball:
11+
arg_groups: []
12+
args: [[10, 20, 40, 100, 200, 400, 1000]]
13+
query_args: []

‎ann_benchmarks/algorithms/datasketch/config.yml‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
bit:
2+
jaccard:
3+
- base_args: ['@metric']
4+
constructor: DataSketch
5+
disabled: false
6+
docker_tag: ann-benchmarks-datasketch
7+
module: ann_benchmarks.algorithms.datasketch
8+
name: datasketch
9+
run_groups:
10+
base:
11+
arg_groups: []
12+
args: [[128, 256, 512], [8, 16, 32, 64, 128]]
13+
query_args: []

‎ann_benchmarks/algorithms/diskann/config.yml‎

Lines changed: 175 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,175 @@
1+
float:
2+
angular:
3+
- base_args: ['@metric']
4+
constructor: Vamana
5+
disabled: false
6+
docker_tag: ann-benchmarks-diskann
7+
module: ann_benchmarks.algorithms.diskann
8+
name: vamana(diskann)
9+
run_groups:
10+
vamana_125_32_1:
11+
arg_groups: []
12+
args: [{alpha: 1, l_build: 125, max_outdegree: 32}]
13+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
14+
vamana_125_32_1-1:
15+
arg_groups: []
16+
args: [{alpha: 1.1, l_build: 125, max_outdegree: 32}]
17+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
18+
vamana_125_32_1-2:
19+
arg_groups: []
20+
args: [{alpha: 1.2, l_build: 125, max_outdegree: 32}]
21+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
22+
vamana_125_64_1:
23+
arg_groups: []
24+
args: [{alpha: 1, l_build: 125, max_outdegree: 64}]
25+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
26+
vamana_125_64_1-1:
27+
arg_groups: []
28+
args: [{alpha: 1.1, l_build: 125, max_outdegree: 64}]
29+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
30+
vamana_125_64_1-2:
31+
arg_groups: []
32+
args: [{alpha: 1.2, l_build: 125, max_outdegree: 64}]
33+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
34+
- base_args: ['@metric']
35+
constructor: VamanaPQ
36+
disabled: false
37+
docker_tag: ann-benchmarks-diskann_pq
38+
module: ann_benchmarks.algorithms.diskann
39+
name: vamana-pq(diskann)
40+
run_groups:
41+
vamana_pq_125_32_1-2_14:
42+
arg_groups: []
43+
args: [{alpha: 1.2, chunks: 14, l_build: 125, max_outdegree: 32}]
44+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
45+
vamana_pq_125_32_1-2_28:
46+
arg_groups: []
47+
args: [{alpha: 1.2, chunks: 28, l_build: 125, max_outdegree: 32}]
48+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
49+
vamana_pq_125_32_1-2_42:
50+
arg_groups: []
51+
args: [{alpha: 1.2, chunks: 42, l_build: 125, max_outdegree: 32}]
52+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
53+
vamana_pq_125_32_1_14:
54+
arg_groups: []
55+
args: [{alpha: 1, chunks: 14, l_build: 125, max_outdegree: 32}]
56+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
57+
vamana_pq_125_32_1_28:
58+
arg_groups: []
59+
args: [{alpha: 1, chunks: 28, l_build: 125, max_outdegree: 32}]
60+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
61+
vamana_pq_125_32_1_42:
62+
arg_groups: []
63+
args: [{alpha: 1, chunks: 42, l_build: 125, max_outdegree: 32}]
64+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
65+
vamana_pq_125_64_1-2_14:
66+
arg_groups: []
67+
args: [{alpha: 1.2, chunks: 14, l_build: 125, max_outdegree: 64}]
68+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
69+
vamana_pq_125_64_1-2_28:
70+
arg_groups: []
71+
args: [{alpha: 1.2, chunks: 28, l_build: 125, max_outdegree: 64}]
72+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
73+
vamana_pq_125_64_1-2_42:
74+
arg_groups: []
75+
args: [{alpha: 1.2, chunks: 42, l_build: 125, max_outdegree: 64}]
76+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
77+
vamana_pq_125_64_1_14:
78+
arg_groups: []
79+
args: [{alpha: 1, chunks: 14, l_build: 125, max_outdegree: 64}]
80+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
81+
vamana_pq_125_64_1_28:
82+
arg_groups: []
83+
args: [{alpha: 1, chunks: 28, l_build: 125, max_outdegree: 64}]
84+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
85+
vamana_pq_125_64_1_42:
86+
arg_groups: []
87+
args: [{alpha: 1, chunks: 42, l_build: 125, max_outdegree: 64}]
88+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
89+
euclidean:
90+
- base_args: ['@metric']
91+
constructor: Vamana
92+
disabled: false
93+
docker_tag: ann-benchmarks-diskann
94+
module: ann_benchmarks.algorithms.diskann
95+
name: vamana(diskann)
96+
run_groups:
97+
vamana_100_64_1:
98+
arg_groups: []
99+
args: [{alpha: 1, l_build: 100, max_outdegree: 64}]
100+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
101+
vamana_100_64_1-1:
102+
arg_groups: []
103+
args: [{alpha: 1.1, l_build: 100, max_outdegree: 64}]
104+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
105+
vamana_100_64_1-2:
106+
arg_groups: []
107+
args: [{alpha: 1.2, l_build: 100, max_outdegree: 64}]
108+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
109+
vamana_125_32_1:
110+
arg_groups: []
111+
args: [{alpha: 1, l_build: 125, max_outdegree: 32}]
112+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
113+
vamana_125_32_1-1:
114+
arg_groups: []
115+
args: [{alpha: 1.1, l_build: 125, max_outdegree: 32}]
116+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
117+
vamana_125_32_1-2:
118+
arg_groups: []
119+
args: [{alpha: 1.2, l_build: 125, max_outdegree: 32}]
120+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
121+
- base_args: ['@metric']
122+
constructor: VamanaPQ
123+
disabled: false
124+
docker_tag: ann-benchmarks-diskann_pq
125+
module: ann_benchmarks.algorithms.diskann
126+
name: vamana-pq(diskann)
127+
run_groups:
128+
vamana_pq_100_64_1-2_32:
129+
arg_groups: []
130+
args: [{alpha: 1.2, chunks: 32, l_build: 100, max_outdegree: 64}]
131+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
132+
vamana_pq_100_64_1_32:
133+
arg_groups: []
134+
args: [{alpha: 1, chunks: 32, l_build: 100, max_outdegree: 64}]
135+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
136+
vamana_pq_125_32_1-2_112:
137+
arg_groups: []
138+
args: [{alpha: 1.2, chunks: 112, l_build: 125, max_outdegree: 32}]
139+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
140+
vamana_pq_125_32_1-2_32:
141+
arg_groups: []
142+
args: [{alpha: 1.2, chunks: 32, l_build: 125, max_outdegree: 32}]
143+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
144+
vamana_pq_125_32_1-2_96:
145+
arg_groups: []
146+
args: [{alpha: 1.2, chunks: 96, l_build: 125, max_outdegree: 32}]
147+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
148+
vamana_pq_125_32_1_112:
149+
arg_groups: []
150+
args: [{alpha: 1, chunks: 112, l_build: 125, max_outdegree: 32}]
151+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
152+
vamana_pq_125_32_1_32:
153+
arg_groups: []
154+
args: [{alpha: 1, chunks: 32, l_build: 125, max_outdegree: 32}]
155+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
156+
vamana_pq_125_32_1_96:
157+
arg_groups: []
158+
args: [{alpha: 1, chunks: 96, l_build: 125, max_outdegree: 32}]
159+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
160+
vamana_pq_80_64_1-2_112:
161+
arg_groups: []
162+
args: [{alpha: 1.2, chunks: 112, l_build: 80, max_outdegree: 64}]
163+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
164+
vamana_pq_80_64_1-2_96:
165+
arg_groups: []
166+
args: [{alpha: 1.2, chunks: 96, l_build: 80, max_outdegree: 64}]
167+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
168+
vamana_pq_80_64_1_112:
169+
arg_groups: []
170+
args: [{alpha: 1, chunks: 112, l_build: 80, max_outdegree: 64}]
171+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]
172+
vamana_pq_80_64_1_96:
173+
arg_groups: []
174+
args: [{alpha: 1, chunks: 96, l_build: 80, max_outdegree: 64}]
175+
query_args: [[10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]]

‎ann_benchmarks/algorithms/dolphinnpy/config.yml‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
float:
2+
any:
3+
- base_args: {}
4+
constructor: DolphinnPy
5+
disabled: true
6+
docker_tag: ann-benchmarks-dolphinn
7+
module: ann_benchmarks.algorithms.dolphinnpy
8+
name: DolphinnPy
9+
run_groups:
10+
base:
11+
arg_groups: []
12+
args: [[10, 50, 100, 200, 1000, 2000]]
13+
query_args: []

‎ann_benchmarks/algorithms/dummy_algo/config.yml‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
float:
2+
any:
3+
- base_args: ['@metric']
4+
constructor: DummyAlgoSt
5+
disabled: true
6+
docker_tag: ann-benchmarks-sklearn
7+
module: ann_benchmarks.algorithms.dummy_algo
8+
name: dummy-algo-st
9+
run_groups:
10+
empty:
11+
arg_groups: []
12+
args: []
13+
query_args: []
14+
- base_args: ['@metric']
15+
constructor: DummyAlgoMt
16+
disabled: true
17+
docker_tag: ann-benchmarks-sklearn
18+
module: ann_benchmarks.algorithms.dummy_algo
19+
name: dummy-algo-mt
20+
run_groups:
21+
empty:
22+
arg_groups: []
23+
args: []
24+
query_args: []

0 commit comments

Comments
 (0)