Skip to content

Commit 838153b

Browse files
committed
add returns to the predict docstring
1 parent f138d13 commit 838153b

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

mne/decoding/base.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -472,12 +472,17 @@ def fit(self, X, y, **fit_params):
472472
return self
473473

474474
def predict(self, X):
475-
"""Predict labels of new data using fitted linear model.
475+
"""Predict class labels for X using fitted linear model.
476476
477477
Parameters
478478
----------
479479
X : array, shape (n_samples, n_features)
480-
Data to label.
480+
The data matrix for which we want to get the predictions.
481+
482+
Returns
483+
-------
484+
y_pred : array, shape (n_samples,)
485+
Vector containing the class labels for each sample.
481486
"""
482487
check_is_fitted(self)
483488
return self.model_.predict(X)

0 commit comments

Comments
 (0)