Function main

knn_utils/saveKNNMulti.py:306–326

()

    return args

def main():
    args = parse_args()
    
    # Create KNNSearchMulti instance
    knn_search = KNNSearchMulti(
        dstore_path=args.dstore_path,
        val_path=args.val_path,
        index_path=args.index_path,
        output_path=args.output_path,
        model_path=args.model_path,
        k=args.k,
        knn_temp=args.knn_temp,
        probe=args.probe,
        batch_size=args.batch_size,
        knn_gpu=args.knn_gpu,
        ignore_first=args.ignore_first,
        threshold=args.threshold
    )
    
    # Process the data
    knn_search.process()

if __name__ == "__main__":
    main()