We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3f12ed8 commit 0a0266fCopy full SHA for 0a0266f
1 file changed
cebra/integrations/sklearn/cebra.py
@@ -600,8 +600,12 @@ def num_sessions(self) -> Optional[int]:
600
return self.num_sessions_
601
602
@property
603
- def num_negatives_(self) -> int:
604
- """The number of negative examples."""
+ def num_negatives_(self) -> Optional[int]:
+ """The number of negative examples.
605
+
606
+ If the model is trained using the full dataset (``batch_size`` is ``None``), this
607
+ function will return ``None``.
608
+ """
609
if self.num_negatives is None:
610
return self.batch_size
611
return self.num_negatives
0 commit comments