milvus-io /
milvus
| @@ 678-721 (lines=44) @@ | ||
| 675 | return search_params |
|
| 676 | ||
| 677 | ||
| 678 | def gen_invalid_index(): |
|
| 679 | index_params = [] |
|
| 680 | for index_type in gen_invalid_strs(): |
|
| 681 | index_param = {"index_type": index_type, "params": {"nlist": 1024}} |
|
| 682 | index_params.append(index_param) |
|
| 683 | for nlist in gen_invalid_params(): |
|
| 684 | index_param = {"index_type": "IVF_FLAT", "params": {"nlist": nlist}} |
|
| 685 | index_params.append(index_param) |
|
| 686 | for M in gen_invalid_params(): |
|
| 687 | index_param = {"index_type": "HNSW", "params": {"M": M, "efConstruction": 100}} |
|
| 688 | index_param = {"index_type": "RHNSW_PQ", "params": {"M": M, "efConstruction": 100}} |
|
| 689 | index_param = {"index_type": "RHNSW_SQ", "params": {"M": M, "efConstruction": 100}} |
|
| 690 | index_params.append(index_param) |
|
| 691 | for efConstruction in gen_invalid_params(): |
|
| 692 | index_param = {"index_type": "HNSW", "params": {"M": 16, "efConstruction": efConstruction}} |
|
| 693 | index_param = {"index_type": "RHNSW_PQ", "params": {"M": 16, "efConstruction": efConstruction}} |
|
| 694 | index_param = {"index_type": "RHNSW_SQ", "params": {"M": 16, "efConstruction": efConstruction}} |
|
| 695 | index_params.append(index_param) |
|
| 696 | for search_length in gen_invalid_params(): |
|
| 697 | index_param = {"index_type": "NSG", |
|
| 698 | "params": {"search_length": search_length, "out_degree": 40, "candidate_pool_size": 50, |
|
| 699 | "knng": 100}} |
|
| 700 | index_params.append(index_param) |
|
| 701 | for out_degree in gen_invalid_params(): |
|
| 702 | index_param = {"index_type": "NSG", |
|
| 703 | "params": {"search_length": 100, "out_degree": out_degree, "candidate_pool_size": 50, |
|
| 704 | "knng": 100}} |
|
| 705 | index_params.append(index_param) |
|
| 706 | for candidate_pool_size in gen_invalid_params(): |
|
| 707 | index_param = {"index_type": "NSG", "params": {"search_length": 100, "out_degree": 40, |
|
| 708 | "candidate_pool_size": candidate_pool_size, |
|
| 709 | "knng": 100}} |
|
| 710 | index_params.append(index_param) |
|
| 711 | index_params.append({"index_type": "IVF_FLAT", "params": {"invalid_key": 1024}}) |
|
| 712 | index_params.append({"index_type": "HNSW", "params": {"invalid_key": 16, "efConstruction": 100}}) |
|
| 713 | index_params.append({"index_type": "RHNSW_PQ", "params": {"invalid_key": 16, "efConstruction": 100}}) |
|
| 714 | index_params.append({"index_type": "RHNSW_SQ", "params": {"invalid_key": 16, "efConstruction": 100}}) |
|
| 715 | index_params.append({"index_type": "NSG", |
|
| 716 | "params": {"invalid_key": 100, "out_degree": 40, "candidate_pool_size": 300, |
|
| 717 | "knng": 100}}) |
|
| 718 | for invalid_n_trees in gen_invalid_params(): |
|
| 719 | index_params.append({"index_type": "ANNOY", "params": {"n_trees": invalid_n_trees}}) |
|
| 720 | ||
| 721 | return index_params |
|
| 722 | ||
| 723 | ||
| 724 | def gen_index(): |
|
| @@ 756-799 (lines=44) @@ | ||
| 753 | return search_params |
|
| 754 | ||
| 755 | ||
| 756 | def gen_invalid_index(): |
|
| 757 | index_params = [] |
|
| 758 | for index_type in gen_invalid_strs(): |
|
| 759 | index_param = {"index_type": index_type, "params": {"nlist": 1024}} |
|
| 760 | index_params.append(index_param) |
|
| 761 | for nlist in gen_invalid_params(): |
|
| 762 | index_param = {"index_type": "IVF_FLAT", "params": {"nlist": nlist}} |
|
| 763 | index_params.append(index_param) |
|
| 764 | for M in gen_invalid_params(): |
|
| 765 | index_param = {"index_type": "HNSW", "params": {"M": M, "efConstruction": 100}} |
|
| 766 | index_param = {"index_type": "RHNSW_PQ", "params": {"M": M, "efConstruction": 100}} |
|
| 767 | index_param = {"index_type": "RHNSW_SQ", "params": {"M": M, "efConstruction": 100}} |
|
| 768 | index_params.append(index_param) |
|
| 769 | for efConstruction in gen_invalid_params(): |
|
| 770 | index_param = {"index_type": "HNSW", "params": {"M": 16, "efConstruction": efConstruction}} |
|
| 771 | index_param = {"index_type": "RHNSW_PQ", "params": {"M": 16, "efConstruction": efConstruction}} |
|
| 772 | index_param = {"index_type": "RHNSW_SQ", "params": {"M": 16, "efConstruction": efConstruction}} |
|
| 773 | index_params.append(index_param) |
|
| 774 | for search_length in gen_invalid_params(): |
|
| 775 | index_param = {"index_type": "NSG", |
|
| 776 | "params": {"search_length": search_length, "out_degree": 40, "candidate_pool_size": 50, |
|
| 777 | "knng": 100}} |
|
| 778 | index_params.append(index_param) |
|
| 779 | for out_degree in gen_invalid_params(): |
|
| 780 | index_param = {"index_type": "NSG", |
|
| 781 | "params": {"search_length": 100, "out_degree": out_degree, "candidate_pool_size": 50, |
|
| 782 | "knng": 100}} |
|
| 783 | index_params.append(index_param) |
|
| 784 | for candidate_pool_size in gen_invalid_params(): |
|
| 785 | index_param = {"index_type": "NSG", "params": {"search_length": 100, "out_degree": 40, |
|
| 786 | "candidate_pool_size": candidate_pool_size, |
|
| 787 | "knng": 100}} |
|
| 788 | index_params.append(index_param) |
|
| 789 | index_params.append({"index_type": "IVF_FLAT", "params": {"invalid_key": 1024}}) |
|
| 790 | index_params.append({"index_type": "HNSW", "params": {"invalid_key": 16, "efConstruction": 100}}) |
|
| 791 | index_params.append({"index_type": "RHNSW_PQ", "params": {"invalid_key": 16, "efConstruction": 100}}) |
|
| 792 | index_params.append({"index_type": "RHNSW_SQ", "params": {"invalid_key": 16, "efConstruction": 100}}) |
|
| 793 | index_params.append({"index_type": "NSG", |
|
| 794 | "params": {"invalid_key": 100, "out_degree": 40, "candidate_pool_size": 300, |
|
| 795 | "knng": 100}}) |
|
| 796 | for invalid_n_trees in gen_invalid_params(): |
|
| 797 | index_params.append({"index_type": "ANNOY", "params": {"n_trees": invalid_n_trees}}) |
|
| 798 | ||
| 799 | return index_params |
|
| 800 | ||
| 801 | ||
| 802 | def gen_index(): |
|
