sklearn.model_selection.train_test_split()

sklearn.model_selection.train_test_split

sklearn.model_selection.train_test_split(*arrays, **options) [source]

Split arrays or matrices into random train and test subsets

Quick utility that wraps input validation and next(ShuffleSplit().split(X, y)) and application to input data into a single call for splitting (and optionally subsampling) data in a oneliner.