Skip to content

Update the docstrings to match the actual code defaults #8701

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ytl0623 wants to merge 1 commit into
base: dev
Choose a base branch
from
Open

Update the docstrings to match the actual code defaults #8701

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 3 additions & 4 deletions monai/losses/unified_focal_loss.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,7 @@ def __init__(
Args:
to_onehot_y : whether to convert `y` into the one-hot format. Defaults to False.
delta : weight of the background. Defaults to 0.7.
gamma : value of the exponent gamma in the definition of the Focal loss . Defaults to 0.75.
gamma : value of the exponent gamma in the definition of the Focal loss . Defaults to 2.
epsilon : it defines a very small number each time. simmily smooth value. Defaults to 1e-7.
"""
super().__init__(reduction=LossReduction(reduction).value)
Expand Down Expand Up @@ -168,9 +168,8 @@ def __init__(
to_onehot_y : whether to convert `y` into the one-hot format. Defaults to False.
num_classes : number of classes, it only supports 2 now. Defaults to 2.
delta : weight of the background. Defaults to 0.7.
gamma : value of the exponent gamma in the definition of the Focal loss. Defaults to 0.75.
epsilon : it defines a very small number each time. simmily smooth value. Defaults to 1e-7.
weight : weight for each loss function, if it's none it's 0.5. Defaults to None.
gamma : value of the exponent gamma in the definition of the Focal loss. Defaults to 0.5.
weight : weight for each loss function. Defaults to 0.5.

Example:
>>> import torch
Expand Down
Loading