When using scikit-learn’s train_test_split, if you want to keep the proportion of classes in the sample the same as the proportion of classes in the entire dataset, use stratify=y.
Split Data in a Stratified Fashion in scikit-learn
Stay up-to-date with
data skills using
CodeCut
CodeCut is a platform that offers short and visually appealing code snippets related to data science, data analysis, data engineering, and Python programming.
Follow Us on Social Media
Copyright © 2025 Code Cut - All rights reserved.