HSALAS
4/19/2018 - 2:36 PM

Split data intro training and validation data

Train 80% Test 20%

snippet.py
content_copyfile_download
#Choosing the Prediction Target
Y = melbourne_data.Price
#Choosing predictors
melbourne_predictors = ['Rooms', 'Bathroom', 'Landsize', 'BuildingArea','YearBuilt', 'Lattitude', 'Longtitude']
X = melbourne_data[melbourne_predictors]
#Split data for training and validation
train_X, test_X, train_Y, test_Y = train_test_split(X, Y,random_state = 0, test_size=0.2)