-
Notifications
You must be signed in to change notification settings - Fork 80
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add vector search param source #425
Conversation
This depends on #424 |
2acd13f
to
b9e453e
Compare
Tested this PR with 1m-128d-l2-hdf5 vector dataset |
84a9651
to
61a71bd
Compare
61a71bd
to
42b7155
Compare
42b7155
to
9c9197d
Compare
9c9197d
to
cd361b8
Compare
osbenchmark/workload/params.py
Outdated
return self.delegate_param_source.partition(partition_index, total_partitions) | ||
|
||
def params(self): | ||
raise exceptions.BenchmarkError("Do not use a VectorSearchParamSource without partitioning") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Might be more appropriate to use exceptions.WorkloadConfigError
as apposed to catch-all exception exceptions.BenchmarkError
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Updated.
Added new param source to partition vector dataset and neighbors. This will be passed to runner to perform search and compare response with neighbors for recall calculation. This param source extends Search ParamSource to inherit search's other query parameters. Vector Param Source will add additional paramter that are required for vector serach operation type. Signed-off-by: Vijayan Balasubramanian <[email protected]>
cd361b8
to
6d09dba
Compare
Description
Added new param source to partition vector dataset and
neighbors. This will be passed to runner to perform
search and compare response with neighbors for recall
calculation.
This param source extends Search ParamSource to inherit search's
other query parameters.
Vector Param Source will add additional paramter that are required
for vector serach operation type.
Issues Resolved
Part of #103
Testing
[Describe how this change was tested]
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.
For more information on following Developer Certificate of Origin and signing off your commits, please check here.