Skip to content

Commit

Permalink
Fix docs
Browse files Browse the repository at this point in the history
  • Loading branch information
stes committed Jun 24, 2023
1 parent f8adf46 commit 59e3970
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions cebra/integrations/sklearn/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,8 +163,9 @@ def _consistency_datasets(
labels: List of labels corresponding to each embedding and to use for alignment
between them.
num_discretization_bins: Number of values for the digitalized common labels. The discretized labels are used
for embedding alignment. Also see the ``n_bins`` argument in :py:mod:`~..helpers.align_embeddings`
for more information on how this parameter is used internally. This argument is only used if ``labels``
for embedding alignment. Also see the ``n_bins`` argument in
:py:mod:`cebra.integrations.sklearn.helpers.align_embeddings` for more information on how this
parameter is used internally. This argument is only used if ``labels``
is not ``None`` and the given labels are continuous and not already discrete.
Returns:
Expand Down Expand Up @@ -327,8 +328,9 @@ def consistency_score(
trained on the **same dataset**. *Consistency between datasets* means the consistency between embeddings
obtained from models trained on **different datasets**, such as different animals, sessions, etc.
num_discretization_bins: Number of values for the digitalized common labels. The discretized labels are used
for embedding alignment. Also see the ``n_bins`` argument in :py:mod:`~..helpers.align_embeddings`
for more information on how this parameter is used internally. This argument is only used if ``labels``
for embedding alignment. Also see the ``n_bins`` argument in
:py:mod:`cebra.integrations.sklearn.helpers.align_embeddings` for more information on how this
parameter is used internally. This argument is only used if ``labels``
is not ``None``, alignment between datasets is used (``between = "datasets"``), and the given labels
are continuous and not already discrete.
Expand Down

0 comments on commit 59e3970

Please sign in to comment.