Update Day 1_Data_Preprocessing.py

This commit is contained in:
wengJJ
2018-09-06 22:00:43 +08:00
committed by GitHub
parent 054df26c99
commit 8b03cb1ced

View File

@ -42,7 +42,7 @@ print("Y")
print(Y)
#Step 5: Splitting the datasets into training sets and Test sets
from sklearn.cross_validation import train_test_split
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 = 0)
print("---------------------")
print("Step 5: Splitting the datasets into training sets and Test sets")