DocumentationClassesStratifiedShuffleSplit

Class: StratifiedShuffleSplit

Stratified ShuffleSplit cross-validator.

Provides train/test indices to split data in train/test sets.

This cross-validation object is a merge of StratifiedKFold and ShuffleSplit, which returns stratified randomized folds. The folds are made by preserving the percentage of samples for each class.

Note: like the ShuffleSplit strategy, stratified random splits do not guarantee that test sets across all folds will be mutually exclusive, and might include overlapping samples. However, this is still very likely for sizeable datasets.

Read more in the User Guide.

For visualisation of cross-validation behaviour and comparison between common scikit-learn split methods refer to Visualizing cross-validation behavior in scikit-learn

Python Reference

Constructors

new StratifiedShuffleSplit()

new StratifiedShuffleSplit(opts?): StratifiedShuffleSplit

Parameters

ParameterTypeDescription
opts?object-
opts.n_splits?numberNumber of re-shuffling & splitting iterations.
opts.random_state?numberControls the randomness of the training and testing indices produced. Pass an int for reproducible output across multiple function calls. See Glossary.
opts.test_size?numberIf float, should be between 0.0 and 1.0 and represent the proportion of the dataset to include in the test split. If int, represents the absolute number of test samples. If undefined, the value is set to the complement of the train size. If train_size is also undefined, it will be set to 0.1.
opts.train_size?numberIf float, should be between 0.0 and 1.0 and represent the proportion of the dataset to include in the train split. If int, represents the absolute number of train samples. If undefined, the value is automatically set to the complement of the test size.

Returns StratifiedShuffleSplit

Defined in generated/model_selection/StratifiedShuffleSplit.ts:31

Properties

PropertyTypeDefault valueDefined in
_isDisposedbooleanfalsegenerated/model_selection/StratifiedShuffleSplit.ts:29
_isInitializedbooleanfalsegenerated/model_selection/StratifiedShuffleSplit.ts:28
_pyPythonBridgeundefinedgenerated/model_selection/StratifiedShuffleSplit.ts:27
idstringundefinedgenerated/model_selection/StratifiedShuffleSplit.ts:24
optsanyundefinedgenerated/model_selection/StratifiedShuffleSplit.ts:25

Accessors

py

Get Signature

get py(): PythonBridge

Returns PythonBridge

Set Signature

set py(pythonBridge): void

Parameters

ParameterType
pythonBridgePythonBridge

Returns void

Defined in generated/model_selection/StratifiedShuffleSplit.ts:58

Methods

dispose()

dispose(): Promise<void>

Disposes of the underlying Python resources.

Once dispose() is called, the instance is no longer usable.

Returns Promise<void>

Defined in generated/model_selection/StratifiedShuffleSplit.ts:114


get_metadata_routing()

get_metadata_routing(opts): Promise<any>

Get metadata routing of this object.

Please check User Guide on how the routing mechanism works.

Parameters

ParameterTypeDescription
optsobject-
opts.routing?anyA MetadataRequest encapsulating routing information.

Returns Promise<any>

Defined in generated/model_selection/StratifiedShuffleSplit.ts:133


get_n_splits()

get_n_splits(opts): Promise<number>

Returns the number of splitting iterations in the cross-validator.

Parameters

ParameterTypeDescription
optsobject-
opts.groups?anyAlways ignored, exists for compatibility.
opts.X?anyAlways ignored, exists for compatibility.
opts.y?anyAlways ignored, exists for compatibility.

Returns Promise<number>

Defined in generated/model_selection/StratifiedShuffleSplit.ts:169


init()

init(py): Promise<void>

Initializes the underlying Python resources.

This instance is not usable until the Promise returned by init() resolves.

Parameters

ParameterType
pyPythonBridge

Returns Promise<void>

Defined in generated/model_selection/StratifiedShuffleSplit.ts:71


split()

split(opts): Promise<ArrayLike>

Generate indices to split data into training and test set.

Parameters

ParameterTypeDescription
optsobject-
opts.groups?anyAlways ignored, exists for compatibility.
opts.X?ArrayLike[]Training data, where n_samples is the number of samples and n_features is the number of features. Note that providing y is sufficient to generate the splits and hence np.zeros(n_samples) may be used as a placeholder for X instead of actual training data.
opts.y?ArrayLikeThe target variable for supervised learning problems. Stratification is done based on the y labels.

Returns Promise<ArrayLike>

Defined in generated/model_selection/StratifiedShuffleSplit.ts:215