@@ -110,12 +110,21 @@ def plot_dataset(dataset_key, dataset_name):
110110 if "jvector" in filename :
111111 algo_name = "JVector"
112112 is_jvector = True
113- log_pattern = f"jvector-*-full_*_memory.log" # Simplified pattern matching
114- # Need to be more specific to match dataset
115- if "euclidean" in dataset_key :
116- log_pattern = "jvector-euclidean-full_*_memory.log"
117- else :
118- log_pattern = "jvector-angular-full_*_memory.log"
113+
114+ # Try matching new explicit key=value format log files
115+ # Pattern: jvector-dataset={dataset_key}_size=full_..._memory.log
116+ # We use glob with wildcards for other params
117+ # Note: Now uses xmx= instead of heap= and includes mutations=
118+ log_pattern = f"jvector-dataset=*{ dataset_key } *_size=full_*_memory.log"
119+
120+ # Fallback for older formats if not found later
121+ if not glob .glob (os .path .join (LOGS_DIR , log_pattern )):
122+ log_pattern = f"jvector-*-full_*_memory.log"
123+ # Need to be more specific to match dataset
124+ if "euclidean" in dataset_key :
125+ log_pattern = "jvector-euclidean-full_*_memory.log"
126+ else :
127+ log_pattern = "jvector-angular-full_*_memory.log"
119128
120129 elif "faiss" in filename :
121130 dataset_type = "euclidean" if "euclidean" in dataset_key else "angular"
@@ -231,11 +240,9 @@ def plot_dataset(dataset_key, dataset_name):
231240 plt .xlim (0.0 , 1.025 )
232241
233242 # Save plot
234- output_png = os .path .join (FIGURES_DIR , f"plot_{ dataset_key } .png" )
235- output_pdf = os .path .join (FIGURES_DIR , f"plot_{ dataset_key } .pdf" )
236- plt .savefig (output_png )
243+ output_pdf = os .path .join (FIGURES_DIR , f"plot_dataset={ dataset_key } .pdf" )
237244 plt .savefig (output_pdf )
238- print (f"Saved plots to { output_png } and { output_pdf } " )
245+ print (f"Saved plot to { output_pdf } " )
239246
240247
241248def main ():
0 commit comments