evalml.preprocessing.split_data

class evalml.preprocessing.split_data

Splits data into train and test sets.

Parameters

• X (DataFrame) – features

• y (Series) – labels

• regression (bool) – if true, do not use stratified split

• test_size (float) – percent of train set to holdout for testing

• random_state (int) – seed for the random number generator

Returns

features and labels each split into train and test sets

Return type

DataFrame, DataFrame, Series, Series