Skip to content

Commit

Permalink
skip docstring for the moemnt
Browse files Browse the repository at this point in the history
  • Loading branch information
glemaitre committed Jan 10, 2025
1 parent a32c62d commit 1133dd0
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions skore/src/skore/sklearn/train_test_split/train_test_split.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,26 +89,27 @@ class labels.
>>> X, y = np.arange(10).reshape((5, 2)), range(5)
>>> # Drop-in replacement for sklearn train_test_split
>>> X_train, X_test, y_train, y_test = train_test_split(X, y,
>>> X_train, X_test, y_train, y_test = train_test_split(X, y, # doctest: +SKIP
... test_size=0.33, random_state=42)
>>> X_train
>>> X_train # doctest: +SKIP
array([[4, 5],
[0, 1],
[6, 7]])
>>> # Explicit X and y, makes detection of problems easier
>>> X_train, X_test, y_train, y_test = train_test_split(X=X, y=y,
>>> X_train, X_test, y_train, y_test = train_test_split(X=X, y=y, # doctest: +SKIP
... test_size=0.33, random_state=42)
>>> X_train
>>> X_train # doctest: +SKIP
array([[4, 5],
[0, 1],
[6, 7]])
>>> # When passing X and y explicitly, X is returned before y
>>> arr = np.arange(10).reshape((5, 2))
>>> arr_train, arr_test, X_train, X_test, y_train, y_test = train_test_split(
>>> splits = train_test_split( # doctest: +SKIP
... arr, y=y, X=X, test_size=0.33, random_state=42)
>>> X_train
>>> arr_train, arr_test, X_train, X_test, y_train, y_test = splits # doctest: +SKIP
>>> X_train # doctest: +SKIP
array([[4, 5],
[0, 1],
[6, 7]])
Expand Down

0 comments on commit 1133dd0

Please sign in to comment.