Train test validation sklearn

# credit to the user of StackExchange in the source link
# set stratify=y in the function arguments for stratified selection
# random_state has been fixed for reproducibility

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test 
	= train_test_split(X, y, test_size=0.2, random_state=1) 

# 0.25 x 0.8 = 0.2
X_train, X_val, y_train, y_val 
    = train_test_split(X_train, y_train, test_size=0.25, random_state=1)

Source: datascience.stackexchange.com

Similar codes

12 votes178 views1 answer
Create new env in anaconda
Python
0 vote169 views1 answer
Knn imputation in r
Python
1 vote267 views1 answer
Maximo numero de variables dentro de un .def python
Python
24 votes175 views3 answers
Python open link in browser
Python
75 votes164 views4 answers
How to check for a substring in python
Python
0 vote204 views1 answer
Pandas split dataframe into chunks with a condition
Python
0 vote160 views1 answer
! LaTeX Error: File `pgf{-}pie.sty' not found.
Python
0 vote197 views1 answer
Use argparse to call function and use argument in function
Python
14 votes164 views2 answers
Exit py console
Python
0 vote173 views1 answer
How to install ffmpeg python heroku
Python