Skip to content

Commit 6c99441

Browse files
committed
make beta mandatory for this loss func
1 parent 3166bb3 commit 6c99441

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

chebai/loss/bce_weighted.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,10 @@ def __init__(
3333
data_extractor = data_extractor.labeled
3434
self.data_extractor = data_extractor
3535

36+
assert (
37+
beta is not None
38+
), f"Beta parameter must be provided if this loss ({self.__class__.__name__}) is used."
39+
3640
# If beta is provided, require a data_extractor.
3741
if self.beta is not None and self.data_extractor is None:
3842
raise ValueError("When 'beta' is set, 'data_extractor' must also be set.")

0 commit comments

Comments
 (0)