mirror of https://github.com/milvus-io/milvus.git
26 lines
745 B
Python
26 lines
745 B
Python
|
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
||
|
|
parser.add_argument(
|
||
|
|
'--dataset',
|
||
|
|
metavar='NAME',
|
||
|
|
help='the dataset to load training points from',
|
||
|
|
default='glove-100-angular',
|
||
|
|
choices=DATASETS.keys())
|
||
|
|
parser.add_argument(
|
||
|
|
"-k", "--count",
|
||
|
|
default=10,
|
||
|
|
type=positive_int,
|
||
|
|
help="the number of near neighbours to search for")
|
||
|
|
parser.add_argument(
|
||
|
|
'--definitions',
|
||
|
|
metavar='FILE',
|
||
|
|
help='load algorithm definitions from FILE',
|
||
|
|
default='algos.yaml')
|
||
|
|
parser.add_argument(
|
||
|
|
'--image-tag',
|
||
|
|
default=None,
|
||
|
|
help='pull image first')
|