mgyigit commited on
Commit
da35485
·
verified ·
1 Parent(s): 99abba4

Update src/saving_utils.py

Browse files
Files changed (1) hide show
  1. src/saving_utils.py +1 -1
src/saving_utils.py CHANGED
@@ -335,7 +335,7 @@ def save_affinity_output(
335
  if 'val_mae_errors' in detail:
336
  affinity_results_df.loc[affinity_results_df['Method'] == method_name, f"mae_{i}"] = detail['val_mae_errors'][i]
337
  if 'validation_corrs' in detail:
338
- affinity_results_df.loc[affinity_results_df['Method'] == method_name, f"correlation_{i}"] = detail['validation_corrs'][i]
339
 
340
  # Save updated DataFrames to CSV
341
  leaderboard_df.to_csv(leaderboard_path, index=False)
 
335
  if 'val_mae_errors' in detail:
336
  affinity_results_df.loc[affinity_results_df['Method'] == method_name, f"mae_{i}"] = detail['val_mae_errors'][i]
337
  if 'validation_corrs' in detail:
338
+ affinity_results_df.loc[affinity_results_df['Method'] == method_name, f"corr_{i}"] = detail['validation_corrs'][i]
339
 
340
  # Save updated DataFrames to CSV
341
  leaderboard_df.to_csv(leaderboard_path, index=False)