mgyigit commited on
Commit
299eef5
·
verified ·
1 Parent(s): 1c760b9

Update src/saving_utils.py

Browse files
Files changed (1) hide show
  1. src/saving_utils.py +1 -1
src/saving_utils.py CHANGED
@@ -321,7 +321,7 @@ def save_affinity_output(
321
  if 'val_mae_errors' in detail:
322
  affinity_results_df.loc[affinity_results_df['Method'] == method_name, f"mae_{i}"] = detail['val_mae_errors'][i]
323
  if 'validation_corrs' in detail:
324
- affinity_results_df.loc[affinity_results_df['Method'] == method_name, f"corr_{i}"] = detail['validation_corrs'][i]
325
 
326
  # Save updated DataFrames to CSV
327
  leaderboard_df.to_csv(leaderboard_path, index=False)
 
321
  if 'val_mae_errors' in detail:
322
  affinity_results_df.loc[affinity_results_df['Method'] == method_name, f"mae_{i}"] = detail['val_mae_errors'][i]
323
  if 'validation_corrs' in detail:
324
+ affinity_results_df.loc[affinity_results_df['Method'] == method_name, f"correlation_{i}"] = detail['validation_corrs'][i]
325
 
326
  # Save updated DataFrames to CSV
327
  leaderboard_df.to_csv(leaderboard_path, index=False)