Skip to content

Commit 6777498

Browse files
committed
Add MAELoss docstrings
1 parent 3d0f71f commit 6777498

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

src/graphnet/training/loss_functions.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,10 @@ def _forward(self, prediction: Tensor, target: Tensor) -> Tensor:
6464

6565

6666
class MAELoss(LossFunction):
67+
"""Mean absolute error loss."""
68+
6769
def _forward(self, prediction: Tensor, target: Tensor) -> Tensor:
70+
"""Implement loss calculation."""
6871
return torch.mean(torch.abs(prediction - target), dim=-1)
6972

7073

0 commit comments

Comments
 (0)