Skip to content
Permalink
0e81a7b148
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?

MssBenchmark/algos.yaml

Go to file
 
 
Cannot retrieve contributors at this time
308 lines (307 sloc) 9.58 KB
Raw Blame
bit:
jaccard:
Bruteforce:
disabled: false
docker-tag: ann-benchmarks-sklearn
singularity-tag: ann-bench-sklearn
module: ann_benchmarks.algorithms.bruteforce
constructor: BruteForceBLAS
base-args: ["@metric"]
run-groups:
base:
args: {}
Balltree(Sklearn):
disabled: false
docker-tag: ann-benchmarks-sklearn
singularity-tag: ann-bench-sklearn
module: ann_benchmarks.algorithms.balltree
constructor: BallTree
base-args: ["@metric"]
run-groups:
ball:
args: [[1, 10, 20, 40, 100, 200, 400, 1000]]
VPtree(Nmslib):
disabled: false
docker-tag: ann-benchmarks-nmslib
singularity-tag: ann-bench-nmslib3
module: ann_benchmarks.algorithms.nmslib
constructor: NmslibReuseIndex
base-args: ["@metric", "Byte", "vptree"]
run-groups:
base:
# When @args is a dictionary, algorithm instances will be generated
# by taking the Cartesian product of all of its values.
arg-groups:
- {"tuneK": 10, "desiredRecall": [0.999, 0.997, 0.995, 0.99, 0.97, 0.95, 0.9, 0.85, 0.8,
0.7, 0.6, 0.5, 0.4, 0.3, 0.2, 0.1, 0.05, 0.02, 0.01]}
- False
Datasketch:
disabled: false
docker-tag: ann-benchmarks-datasketch
singularity-tag: ann-bench-datasketch
module: ann_benchmarks.algorithms.datasketch
constructor: DataSketch
base-args: ["@metric"]
run-groups:
base:
args: [[32, 64, 128, 256, 512, 1024, 2048],[10, 30, 50, 70]]
Hnsw(Nmslib):
disabled: false
docker-tag: ann-benchmarks-nmslib
singularity-tag: ann-bench-nmslib3
module: ann_benchmarks.algorithms.nmslib
constructor: NmslibReuseIndex
base-args: ["@metric", "Byte", "hnsw"]
run-groups:
M-48:
arg-groups:
- {"M": 48, "post": 2, "efConstruction": 800}
- False
query-args: [[50, 70, 90, 120, 160, 200, 400, 600, 700, 800, 1000,
1400, 1600, 2000]]
M-32:
arg-groups:
- {"M": 32, "post": 2, "efConstruction": 800}
- False
query-args: [[100, 300, 500, 700, 1000, 1500, 2000]]
M-20:
arg-groups:
- {"M": 20, "post": 0, "efConstruction": 800}
- False
query-args: [[2, 5, 10, 15, 20, 30, 40, 50, 70, 80]]
M-12:
arg-groups:
- {"M": 12, "post": 0, "efConstruction": 800}
- False
query-args: [[1, 2, 5, 10, 15, 20, 30, 40, 50, 70, 80]]
M-5:
arg-groups:
- {"M": 5, "post": 0, "efConstruction": 10}
- False
query-args: [[1, 2, 5, 10]]
M-2:
arg-groups:
- {"M": 2, "post": 0, "efConstruction": 1}
- False
query-args: [[1, 2]]
SW-graph(Nmslib):
disabled: false
docker-tag: ann-benchmarks-nmslib
singularity-tag: ann-bench-nmslib3
module: ann_benchmarks.algorithms.nmslib
constructor: NmslibReuseIndex
base-args: ["@metric", "Byte", "sw-graph"]
run-groups:
NN-96:
arg-groups:
- {"NN": 96}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-48:
arg-groups:
- {"NN": 48}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-24:
arg-groups:
- {"NN": 24}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-16:
arg-groups:
- {"NN": 16}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-10:
arg-groups:
- {"NN": 10}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-5:
arg-groups:
- {"NN": 5}
- False
query-args: [[30, 25, 20, 15, 10, 5, 4, 3, 2, 1]]
NN-2:
arg-groups:
- {"NN": 2}
- False
query-args: [[30, 25, 20, 15, 10, 5, 4, 3, 2, 1]]
NN-1:
arg-groups:
- {"NN": 1}
- False
query-args: [[30, 25, 20, 15, 10, 5, 4, 3, 2, 1]]
Pynndescent:
disabled: false
docker-tag: ann-benchmarks-pynndescent
singularity-tag: ann-bench-pynndescent
module: ann_benchmarks.algorithms.pynndescent
constructor: PyNNDescent
base-args: ["@metric"]
run-groups:
pynndescent:
args: [[2, 5, 10, 20, 40, 80, 120, 160], [2, 4, 8], [30]]
query-args: [[1.0, 2.0, 4.0, 8.0]]
Onng(Ngt):
disabled: false
docker-tag: ann-benchmarks-ngt
singularity-tag: ann-bench-ngt
module: ann_benchmarks.algorithms.onng_ngt
constructor: ONNG
base-args: ["@metric", "Byte", 1.0]
run-groups:
onng:
args: [[100, 300, 500, 1000], [10, 30, 50, 100], [10, 30, 50, 120]]
query-args: [[0.25, 0.5, 0.75, 1.0, 1.25, 1.5, 1.75, 2.0]]
Panng(Ngt):
disabled: false
docker-tag: ann-benchmarks-ngt
singularity-tag: ann-bench-ngt
module: ann_benchmarks.algorithms.panng_ngt
constructor: PANNG
base-args: ["@metric", "Byte"]
run-groups:
panng:
args: [[10, 20, 40], [40], [30, 60, 120]]
query-args: [[0.05, 0.1, 0.2, 0.4, 0.6, 0.8, 0.9, 0.95, 1.0, 1.02, 1.05, 1.1, 1.2, 1.5, 2.0]]
Risc:
disabled: false
docker-tag: ann-benchmarks-risc
singularity-tag: ann-bench-risc
module: ann_benchmarks.algorithms.risc
constructor: Risc
base-args: ["@metric", "Risc"]
run-groups:
empty:
args: []
DivideSkip:
disabled: false
docker-tag: ann-benchmarks-risc
singularity-tag: ann-bench-risc
module: ann_benchmarks.algorithms.risc
constructor: Risc
base-args: ["@metric", "DivideSkip"]
run-groups:
empty:
args: []
int:
jaccard:
Bruteforce:
disabled: false
docker-tag: ann-benchmarks-sklearn
singularity-tag: ann-bench-sklearn
module: ann_benchmarks.algorithms.bruteforce
constructor: BruteForceBLAS
base-args: ["@metric"]
run-groups:
base:
args: {}
Hnsw(Nmslib):
disabled: false
docker-tag: ann-benchmarks-nmslib
singularity-tag: ann-bench-nmslib3
module: ann_benchmarks.algorithms.nmslib
constructor: NmslibReuseIndex
base-args: ["@metric", "Int", "hnsw"]
run-groups:
M-48:
arg-groups:
- {"M": 48, "post": 2, "efConstruction": 800}
- False
query-args: [[50, 70, 90, 120, 160, 200, 400, 600, 700, 800, 1000,
1400, 1600, 2000]]
M-32:
arg-groups:
- {"M": 32, "post": 2, "efConstruction": 800}
- False
query-args: [[100, 300, 500, 700, 1000, 1500, 2000]]
M-20:
arg-groups:
- {"M": 20, "post": 0, "efConstruction": 800}
- False
query-args: [[2, 5, 10, 15, 20, 30, 40, 50, 70, 80]]
M-12:
arg-groups:
- {"M": 12, "post": 0, "efConstruction": 800}
- False
query-args: [[1, 2, 5, 10, 15, 20, 30, 40, 50, 70, 80]]
M-5:
arg-groups:
- {"M": 5, "post": 0, "efConstruction": 10}
- False
query-args: [[1, 2, 5, 10]]
M-2:
arg-groups:
- {"M": 2, "post": 0, "efConstruction": 1}
- False
query-args: [[1, 2]]
SW-graph(Nmslib):
disabled: false
docker-tag: ann-benchmarks-nmslib
singularity-tag: ann-bench-nmslib3
module: ann_benchmarks.algorithms.nmslib
constructor: NmslibReuseIndex
base-args: ["@metric", "Int", "sw-graph"]
run-groups:
NN-96:
arg-groups:
- {"NN": 96}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-48:
arg-groups:
- {"NN": 48}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-24:
arg-groups:
- {"NN": 24}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-16:
arg-groups:
- {"NN": 16}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-10:
arg-groups:
- {"NN": 10}
- False
query-args: [[800, 400, 200, 100, 50, 30, 20, 15, 10, 5, 1]]
NN-5:
arg-groups:
- {"NN": 5}
- False
query-args: [[30, 25, 20, 15, 10, 5, 4, 3, 2, 1]]
NN-2:
arg-groups:
- {"NN": 2}
- False
query-args: [[30, 25, 20, 15, 10, 5, 4, 3, 2, 1]]
NN-1:
arg-groups:
- {"NN": 1}
- False
query-args: [[30, 25, 20, 15, 10, 5, 4, 3, 2, 1]]
Onng(Ngt):
disabled: false
docker-tag: ann-benchmarks-ngt
singularity-tag: ann-bench-ngt
module: ann_benchmarks.algorithms.onng_ngt
constructor: ONNG
base-args: ["@metric", "Byte", 1.0]
run-groups:
onng:
args: [[100, 300, 500, 1000], [10, 30, 50, 100], [10, 30, 50, 120]]
query-args: [[0.25, 0.5, 0.75, 1.0, 1.25, 1.5, 1.75, 2.0]]
Risc:
disabled: false
docker-tag: ann-benchmarks-risc
singularity-tag: ann-bench-risc
module: ann_benchmarks.algorithms.risc
constructor: Risc
base-args: ["@metric", "Risc"]
run-groups:
empty:
args: []