Skip to content

Commit

Permalink
docs(website): generalize name to "Data splitting" (#128)
Browse files Browse the repository at this point in the history
  • Loading branch information
deepyaman authored Jul 19, 2024
1 parent 6041907 commit 722c448
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 11 deletions.
10 changes: 5 additions & 5 deletions docs/_quarto.yml
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ website:

- section: Utilities
contents:
- reference/utils-train-test-split.qmd
- reference/utils-data-splitting.qmd

format:
html:
Expand Down Expand Up @@ -204,7 +204,7 @@ quartodoc:
path: steps-outlier-handling
summary:
name: Outlier handling
desc: Handle outliers
desc: Outlier detection and handling
contents:
- HandleUnivariateOutliers

Expand Down Expand Up @@ -234,9 +234,9 @@ quartodoc:
package: ibis_ml
contents:
- kind: page
path: utils-train-test-split
path: utils-data-splitting
summary:
name: Train-test split
desc: Randomly split Ibis table
name: Data splitting
desc: Segregating data into training, testing, and validation sets
contents:
- train_test_split
2 changes: 1 addition & 1 deletion ibis_ml/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
)
from ibis_ml.steps import *
from ibis_ml.utils._pprint import _pprint_recipe, _pprint_step, _safe_repr
from ibis_ml.utils._train_test_split import train_test_split
from ibis_ml.utils._split import train_test_split

# Add support for `Recipe`s and `Step`s to the built-in `PrettyPrinter`.
pprint.PrettyPrinter._dispatch[Recipe.__repr__] = _pprint_recipe # noqa: SLF001
Expand Down
10 changes: 5 additions & 5 deletions ibis_ml/utils/_train_test_split.py → ibis_ml/utils/_split.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,11 +62,11 @@ def train_test_split(
>>> table = ibis.memtable({"key1": range(100)})
>>> train_table, test_table = ml.train_test_split(
table,
unique_key="key1",
test_size=0.2,
random_seed=0,
)
... table,
... unique_key="key1",
... test_size=0.2,
... random_seed=0,
... )
"""
if not (0 < test_size < 1):
raise ValueError("test size should be a float between 0 and 1.")
Expand Down

0 comments on commit 722c448

Please sign in to comment.