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

2 votes305 views1 answer
Cos in python in degrees
Python
1 vote314 views1 answer
Export python pandas dataframe as json file
Python
0 vote295 views1 answer
Wkhtmltopdf pdfkit blocked access to file
Python
3 votes306 views3 answers
Hide turtle
Python
16 votes292 views2 answers
Python removing \n from string
Python
0 vote342 views1 answer
Python cv2 convert image to binary
Python
0 vote310 views1 answer
Python series to list of string
Python
2 votes338 views1 answer
Tf.reduce_sum()
Python
34 votes306 views2 answers
Python create nested directory
Python
0 vote300 views1 answer
Argmax change dafault value for multiple maxima
Python