Link Search Menu Expand Document

In this post we are going to see, Decision Tree - Regression
R Implementation.
rm(list=ls())
setwd()
library(rpart)
library(mass) df = birthwt
View(birthwt)
Here I am loading birthweight Data for Regression analysis as we are going to
predict continuous value. The required library has been uploaded. See below for screenshots..

See below for collecting sample and extracting training and test data

data_index = sample(1:nrow(df), 0.8 * nrow(df))
View(data_index)
training_data = df[data_index,]
View(training_data)
test_data = df[-data_index,]
View(test_data)

See below for screenshots..


We are going to prepare ML Model. See below..

ML_Model = rpart(bwt ~ ., data = training_data, method = “anova”)
View(ML_Model)


After building Machine Learning Model, We apply the model on the test data.
test_predictions = predict(ML_Model, test_data[,-10])

There are different error metrics to measure the error rate. Below are some..
Mean Squared Error (MSE)
Root Mean Squared Error (RMSE)
Mean Absolute Error (MAE)
Mean absolute percentage error (MAPE)

Below is the method to calculate

regr.eval(test_data[,10], test_predictions, stats = c(‘mae’,’rmse’,’mape’, ‘mse’))

in the above screenshot its calculated. For mape it says 12 % error rate. It is mentioned in exponential where we have to convert to %..

Hope its clear!..

Lets move next!!!!