From 4e4d42a28179b5a07c9a19620fa7652315c7394f Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 12 Dec 2022 18:34:20 +0000 Subject: [PATCH] LossDice: add comment --- aimodel/src/lib/ai/components/LossDice.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aimodel/src/lib/ai/components/LossDice.py b/aimodel/src/lib/ai/components/LossDice.py index 2f5c0c5..0df4baa 100644 --- a/aimodel/src/lib/ai/components/LossDice.py +++ b/aimodel/src/lib/ai/components/LossDice.py @@ -35,6 +35,7 @@ def dice_coef_loss(y_true, y_pred, **kwargs): class LossDice(tf.keras.losses.Loss): """An implementation of the dice loss function. + @source Args: smooth (float): The batch size (currently unused). """