Train-Test Splitting with Stratification using Scikit-Learn

Python

Ensures that the test and train splits have the same ratio of class ratio for training classification models. We use the stratify parameter and pass the y series.

 1|  from sklearn.model_selection import train_test_split
 2|  X_train, X_test, y_train, y_test = train_test_split(X, 
 3|                                                      y, 
 4|                                                      test_size=0.4, 
 5|                                                      random_state=101,
 6|                                                      stratify=y)
detro
Sklearn | Classification
Did you find this snippet useful?

Sign up for free to to add this to your code library

Remove Stop Words from Text in DataFrame Column
Python

Pandas | Nltk | Nlp

133
LightGBM Custom Loss Function
Python

LightGBM | Loss function

121
Tuning XGBoost Hyperparameters with Grid Search
Python

XGBoost | Tuning

117
How to Convert DataFrame Values Into Percentages
Python

Lambda | Percentages | Pipe

109
How to Scale Data Using Standard Scaler But Keep Column Names
Python

Scaler | Standard

79
LightGBM Hyperparameter Tuning with GridSearch
Python

Python | LightGBM | Hyperparameter tuning | Gridsearch

77
Calculating Root Mean Squared Error (RMSE) with Sklearn and Python
Python

Rmse

77
Dynamically Create Columns in Pandas Dataframe
Python

Pandas | Columns

61
How to Train a Catboost Classifier with GridSearch Hyperparameter Tuning
Python

Catboost | Hyperparameter tuning

60
Overwriting a Python Logging File Instead of Appending
Python

Logging

55