IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-20 11:38:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

adding vectormath for pinecone adapter

Browse Source
This commit is contained in:
Mark Wolters 2023-08-29 15:04:29 -04:00
parent a0c7687cf9
commit c0619e2c1a
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
adapter-pinecone/src/main/resources/activities/pinecone-verify.yaml
View File

@ -18,15 +18,16 @@ blocks:
include_metadata: false include_metadata: false
verifier-imports: verifier-imports:
- "io.nosqlbench.api.engine.metrics.ActivityMetrics" - "io.nosqlbench.api.engine.metrics.ActivityMetrics"
- "io.nosqlbench.engine.extensions.vectormath.Intersections" - "io.nosqlbench.engine.extensions.vectormath.PineconeIntersections"
- "io.nosqlbench.engine.extensions.vectormath.PineconeVectorMath"
verifier-init: | verifier-init: |
recallHisto = ActivityMetrics.histogram(_parsed_op,"recall-histo",4); recallHisto = ActivityMetrics.histogram(_parsed_op,"recall-histo",4);
verifier: | verifier: |
found_string_ids=pinecone_vectormath.idsToStringArray(result); found_string_ids=PineconeVectorMath.idsToStringArray(result);
found_int_ids=pinecone_vectormath.stringArrayAsIntArray(found_string_ids); found_int_ids=PineconeVectorMath.stringArrayAsIntArray(found_string_ids);
print(found_int_ids) print(found_int_ids)
print({validation_set}) print({validation_set})
recall = pinecone_vectormath.computeRecall(found_int_ids, {validation_set}) recall = PineconeVectorMath.computeRecall(found_int_ids, {validation_set})
print(recall) print(recall)
recallHisto.update((long)(recall*1000000.0)); recallHisto.update((long)(recall*1000000.0));
return true; return true;