File tree Expand file tree Collapse file tree 1 file changed +18
-1
lines changed
Expand file tree Collapse file tree 1 file changed +18
-1
lines changed Original file line number Diff line number Diff line change @@ -114,7 +114,6 @@ def export_to_csv(
114114 "update_many" ,
115115 "get_all" ,
116116 "delete_many" ,
117- "similarity_search" ,
118117 ]
119118 for op in operations :
120119 matching = [r for r in mode_results [size ] if r ["operation" ] == op ]
@@ -125,4 +124,22 @@ def export_to_csv(
125124 else :
126125 writer .writerow ([op , "N/A" , "N/A" ])
127126
127+ # Add similarity search results with top_k in operation name
128+ search_results = [
129+ r
130+ for r in mode_results [size ]
131+ if r ["operation" ] == "similarity_search"
132+ ]
133+ for result in search_results :
134+ top_k = result .get ("top_k" , 0 )
135+ ops_per_sec = result .get ("ops_per_sec" , 0 )
136+ time_val = result .get ("time" , 0 )
137+ writer .writerow (
138+ [
139+ f"similarity_search_{ top_k } " ,
140+ f"{ ops_per_sec :.2f} " ,
141+ f"{ time_val :.4f} " ,
142+ ]
143+ )
144+
128145 print (f"Exported { db_mode } ({ size } records) to: { filename } " )
You can’t perform that action at this time.
0 commit comments