HSALAS
4/19/2018 - 2:36 PM

Decision Trees \ Random Forest

#DECISION TREES
# Define model
decisionTrees_model = DecisionTreeRegressor()
# Fit model
decisionTrees_model.fit(train_X, train_Y)
#Get predicted prices on validation data and print MAE
print("Mean Absolute Error:","\n")
print(mean_absolute_error(test_Y, melbourne_model.predict(test_X)))


#RANDOM FOREST
# Define model
randomForest_model = RandomForestRegressor()
# Fit model
randomForest_model.fit(train_X, train_Y)
#Get predicted prices on validation data and print MAE
print("Mean Absolute Error:","\n")
print(mean_absolute_error(test_Y, melbourne_model.predict(test_X)))