Train-test split with 20% of data set aside - ML

Question

I have an assignment of Making a train -test split with 20% of the data set aside for testing. How do I do that using python?

Answer 1

Hi@Kim,

You can use the train_test_split function to do this. You can use the below code in your program.

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=42)

I hope this will help.