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

5 votes177 views2 answers
Check if something is nan python
Python
10 votes215 views3 answers
Set value based on column
Python
0 vote237 views1 answer
Tkinter e.delete(0,END)
Python
0 vote180 views1 answer
Var colors python
Python
2 votes187 views2 answers
Python array get index
Python
15 votes206 views5 answers
Join on column pandas
Python
0 vote192 views1 answer
Python split paragraph
Python
2 votes189 views1 answer
Python cv2 screen capture
Python
0 vote255 views1 answer
Django-filter field name greater than
Python
0 vote268 views1 answer
Speak by a discord bot in python
Python