- 1.22.0 (latest)
- 1.21.0
- 1.20.0
- 1.19.0
- 1.18.0
- 1.17.0
- 1.16.0
- 1.15.0
- 1.14.0
- 1.13.0
- 1.12.0
- 1.11.1
- 1.10.0
- 1.9.0
- 1.8.0
- 1.7.0
- 1.6.0
- 1.5.0
- 1.4.0
- 1.3.0
- 1.2.0
- 1.1.0
- 1.0.0
- 0.26.0
- 0.25.0
- 0.24.0
- 0.23.0
- 0.22.0
- 0.21.0
- 0.20.1
- 0.19.2
- 0.18.0
- 0.17.0
- 0.16.0
- 0.15.0
- 0.14.1
- 0.13.0
- 0.12.0
- 0.11.0
- 0.10.0
- 0.9.0
- 0.8.0
- 0.7.0
- 0.6.0
- 0.5.0
- 0.4.0
- 0.3.0
- 0.2.0
Functions for test/train split and model tuning. This module is styled after scikit-learn's model_selection module: https://scikit-learn.org/stable/modules/classes.html#module-sklearn.model_selection.
Modules Functions
train_test_split
train_test_split(
*arrays: typing.Union[bigframes.dataframe.DataFrame, bigframes.series.Series],
test_size: typing.Optional[float] = None,
train_size: typing.Optional[float] = None,
random_state: typing.Optional[int] = None
) -> typing.List[typing.Union[bigframes.dataframe.DataFrame, bigframes.series.Series]]
Splits dataframes or series into random train and test subsets.
Parameters | |
---|---|
Name | Description |
\*arrays |
bigframes.dataframe.DataFrame or bigframes.series.Series
A sequence of BigQuery DataFrames or Series that can be joined on their indexes |
test_size |
default None
The proportion of the dataset to include in the test split. If None, this will default to the complement of train_size. If both are none, it will be set to 0.25. |
train_size |
default None
The proportion of the dataset to include in the train split. If None, this will default to the complement of test_size. |
random_state |
default None
A seed to use for randomly choosing the rows of the split. If not set, a random split will be generated each time. |
Returns | |
---|---|
Type | Description |
List[Union[bigframes.dataframe.DataFrame, bigframes.series.Series]] | A list of BigQuery DataFrames or Series. |