Skip to content

Commit

Permalink
add clarifying comment
Browse files Browse the repository at this point in the history
  • Loading branch information
valosekj committed Dec 10, 2024
1 parent 4756055 commit f969360
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion compute_metrics_reloaded.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,7 @@ def get_images(prediction, reference):
df = pd.merge(df_pred, df_ref, on=['participant_id', 'acq_id', 'run_id'], how='outer', suffixes=('_pred', '_ref'))
# Drop 'participant_id', 'acq_id', 'run_id'
df.drop(['participant_id', 'acq_id', 'run_id'], axis=1, inplace=True)
# Drop rows with NaN values
# Drop rows with NaN values. In other words, keep only the rows where both prediction and reference files exist
df.dropna(inplace=True)

prediction_files = df['filename_pred'].tolist()
Expand Down

0 comments on commit f969360

Please sign in to comment.