We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5a8de38 commit 6b4343eCopy full SHA for 6b4343e
cebra/integrations/sklearn/metrics.py
@@ -117,9 +117,7 @@ def goodness_of_fit(model: cebra_sklearn_cebra.CEBRA) -> List[float]:
117
different batch sizes or different implementations.
118
119
Args:
120
- model: The model to evaluate. This can be an instance of either
121
- `cebra_sklearn_cebra.CEBRA` or `cebra_solver.Solver`.
122
- batch_size: Batch size used to train the model.
+ model: The model to evaluate. This is an instance of `cebra_sklearn_cebra.CEBRA`.
123
124
Returns:
125
A list of float values representing the goodness of fit for the model.
0 commit comments