Anisha Bhatnagar commited on
Commit
f6912cd
·
1 Parent(s): ce95080

adding model_name to caching

Browse files
utils/interp_space_utils.py CHANGED
@@ -828,7 +828,7 @@ def compute_predicted_author(task_authors_df: pd.DataFrame, col_name: str) -> in
828
  return predicted_author
829
 
830
 
831
- def compute_precomputed_regions(bg_proj, bg_ids, q_proj, c_proj, mystery_id, candidate_ids, n_neighbors=7):
832
  """
833
  Compute precomputed regions for mystery author and candidates.
834
 
@@ -843,7 +843,7 @@ def compute_precomputed_regions(bg_proj, bg_ids, q_proj, c_proj, mystery_id, can
843
  dict: mapping region names to bounding boxes and author lists
844
  """
845
  print("Computing sugested regions for zoom...")
846
- key = f"{hashlib.md5((str(q_proj.tolist()) + str(c_proj.tolist()) + str(n_neighbors)).encode()).hexdigest()}"
847
 
848
  if os.path.exists(REGION_CACHE):
849
  with open(REGION_CACHE, 'rb') as f:
 
828
  return predicted_author
829
 
830
 
831
+ def compute_precomputed_regions(bg_proj, bg_ids, q_proj, c_proj, model_name, n_neighbors=7):
832
  """
833
  Compute precomputed regions for mystery author and candidates.
834
 
 
843
  dict: mapping region names to bounding boxes and author lists
844
  """
845
  print("Computing sugested regions for zoom...")
846
+ key = f"{hashlib.md5((model_name + str(q_proj.tolist()) + str(c_proj.tolist()) + str(n_neighbors)).encode()).hexdigest()}"
847
 
848
  if os.path.exists(REGION_CACHE):
849
  with open(REGION_CACHE, 'rb') as f:
utils/visualizations.py CHANGED
@@ -519,7 +519,7 @@ def visualize_clusters_plotly(iid, cfg, instances, model_radio, custom_model_inp
519
  candidate_ids = task_authors_df['authorID'].iloc[1:4].tolist() # 3 candidate IDs
520
 
521
  precomputed_regions = compute_precomputed_regions(
522
- bg_proj_for_regions, bg_ids_for_regions, q_proj, c_proj, mystery_id, candidate_ids
523
  )
524
 
525
  # Create choices for radio buttons
 
519
  candidate_ids = task_authors_df['authorID'].iloc[1:4].tolist() # 3 candidate IDs
520
 
521
  precomputed_regions = compute_precomputed_regions(
522
+ bg_proj_for_regions, bg_ids_for_regions, q_proj, c_proj, model_name
523
  )
524
 
525
  # Create choices for radio buttons