Permalink
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 fileThis commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
…rce, and optimize the code
331 lines (327 sloc)
10.1 KB
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
bit: | |
jaccard: | |
Bruteforce: | |
disabled: false | |
docker-tag: ann-benchmarks-chemfp | |
singularity-tag: ann-bench-chemfp | |
module: ann_benchmarks.algorithms.bruteforce | |
constructor: BruteForceFPS | |
base-args: ["@metric"] | |
run-groups: | |
empty: | |
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: [] | |
Chemfp: | |
disabled: false | |
docker-tag: ann-benchmarks-chemfp | |
singularity-tag: ann-bench-chemfp | |
module: ann_benchmarks.algorithms.chemfp | |
constructor: Chemfp | |
base-args: ["@metric"] | |
run-groups: | |
empty: | |
args: [] | |
Folding: | |
disabled: false | |
docker-tag: ann-benchmarks-chemfp | |
singularity-tag: ann-bench-chemfp | |
module: ann_benchmarks.algorithms.folding | |
constructor: Folding | |
base-args: ["@metric"] | |
run-groups: | |
panng: | |
args: [[64, 128, 192, 256, 512]] | |
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: [] |