Skip to content

Commit

Permalink
Remove unnecessary wrapping of MAE metric function inside a partial f…
Browse files Browse the repository at this point in the history
…unction
  • Loading branch information
nathanpainchaud committed Nov 6, 2023
1 parent 170ac69 commit 104312a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion didactic/tasks/cardiac_multimodal_representation.py
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,7 @@ def __init__(
) # Hyperparameter to easily access target attributes
for attr in self.predict_losses:
if attr in ClinicalAttribute.numerical_attrs():
self.metrics[attr] = {"mae": functools.partial(mean_absolute_error)}
self.metrics[attr] = {"mae": mean_absolute_error}
elif attr in ClinicalAttribute.binary_attrs():
self.metrics[attr] = {"acc": functools.partial(accuracy, task="binary")}
else: # attr in ClinicalAttribute.categorical_attrs()
Expand Down

0 comments on commit 104312a

Please sign in to comment.