How to Do One Hot Encoding in R
Introduction
There are several powerful machine learning algorithms in R. However, to make the best use of these algorithms, it is imperative that we transform the data into the desired format. One of the common steps for doing this is encoding the data, which enhances the computational power and the efficiency of the algorithms. In this guide, you will learn about the different techniques of encoding data with R.
Data
In this guide, we will use a fictitious dataset of loan applications containing 600 observations and 10 variables:
-
Marital_status: Whether the applicant is married ("Yes") or not ("No") -
Dependents: Number of dependents of the applicant -
Is_graduate: Whether the applicant is a graduate ("Yes") or not ("No") -
Income: Annual Income of the applicant (in USD) -
Loan_amount: Loan amount (in USD) for which the application was submitted -
Credit_score: Whether the applicant's credit score is good ("Satisfactory") or not ("Not Satisfactory") -
Approval_status: Whether the loan application was approved ("1") or not ("0") -
Age: The applicant's age in years -
Sex: Whether the applicant is a male ("M") or a female ("F") -
Purpose: Purpose of applying for the loan
Let's start by loading the required libraries and the data.
1 library(plyr) 2library(readr) 3library(dplyr) 4library(caret) 5 6dat <- read_csv("data_eng.csv") 7 8glimpse(dat) {r}
1 Observations: 600 2Variables: 10 3$ Marital_status <chr> "Yes", "No", "Yes", "Yes", "Yes", "Yes", "Yes", "Yes",... 4$ Dependents <int> 1, 0, 0, 1, 2, 0, 1, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0, ... 5$ Is_graduate <chr> "No", "Yes", "Yes", "Yes", "Yes", "Yes", "Yes", "Yes",... 6$ Income <int> 298500, 315500, 295100, 319300, 333300, 277700, 332100... 7$ Loan_amount <int> 71000, 75500, 70000, 70000, 98000, 71000, 58000, 64000... 8$ Credit_score <chr> "Satisfactory", "Satisfactory", "Satisfactory", "Satis... 9$ approval_status <int> 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... 10$ Age <int> 74, 71, 71, 68, 64, 64, 63, 61, 60, 59, 56, 55, 54, 54... 11$ Sex <chr> "M", "M", "M", "M", "M", "M", "M", "M", "M", "M", "M",... 12$ Purpose <chr> "Wedding", "Wedding", "Wedding", "Wedding", "Wedding",... The output shows that the dataset has six numerical variables (labeled as int ), and four categorical variables (labelled as chr ). We are now ready to carry out the encoding steps.
There are different methods for encoding categorical variables, and selection depends on the distribution of labels in the variable and the end objective. In the subsequent sections, we will cover the most widely used techniques of encoding categorical variables.
Label Encoding
In simple terms, label encoding is the process of replacing the different levels of a categorical variable with dummy numbers. For instance, the variable Credit_score has two levels, "Satisfactory" and "Not_satisfactory". These can be encoded to 1 and 0, respectively. The first line of code below performs this task, while the second line prints a table of the levels post-encoding.
1 dat$Credit_score <- ifelse(dat$Credit_score == "Satisfactory",1,0) 2 3table(dat$Credit_score) {r}
The above output shows that the label encoding is done. This is easy when you have two levels in the categorical variable, as with Credit_score . If the variable contains more than two labels, this will not be intuitive. For example, the 'Purpose' variable has six levels, as can be seen from the output below.
1 Business Education Furniture Personal Travel Wedding 2 43 191 38 166 123 39 In such cases, one-hot encoding is preferred.
One-Hot Encoding
In this technique, one-hot (dummy) encoding is applied to the features, creating a binary column for each category level and returning a sparse matrix. In each dummy variable, the label "1" will represent the existence of the level in the variable, while the label "0" will represent its non-existence.
We will apply this technique to all the remaining categorical variables. The first line of code below imports the powerful caret package, while the second line uses the dummyVars() function to create a full set of dummy variables. The dummyVars() method works on the categorical variables. It is to be noted that the second line contains the argument fullrank=T , which will create n-1 columns for a categorical variable with n unique levels.
The third line uses the output of the dummyVars() function and transforms the dataset, dat , where all the categorical variables are encoded to numerical variables. The fourth line of code prints the structure of the resulting data, dat-transfored , which confirms that one-hot encoding is completed.
1 library(caret) 2 3dmy <- dummyVars(" ~ .", data = dat, fullRank = T) 4dat_transformed <- data.frame(predict(dmy, newdata = dat)) 5 6glimpse(dat_transformed) {r}
1 Observations: 600 2Variables: 14 3$ Marital_status.Yes <dbl> 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, ... 4$ Dependents <dbl> 1, 0, 0, 1, 2, 0, 1, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, ... 5$ Is_graduate.Yes <dbl> 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 0, ... 6$ Income <dbl> 298500, 315500, 295100, 319300, 333300, 277700, 332... 7$ Loan_amount <dbl> 71000, 75500, 70000, 70000, 98000, 71000, 58000, 64... 8$ Credit_score <dbl> 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, ... 9$ approval_status.1 <dbl> 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... 10$ Age <dbl> 74, 71, 71, 68, 64, 64, 63, 61, 60, 59, 56, 55, 54,... 11$ Sex.M <dbl> 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... 12$ Purpose.Education <dbl> 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... 13$ Purpose.Furniture <dbl> 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... 14$ Purpose.Personal <dbl> 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... 15$ Purpose.Travel <dbl> 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ... 16$ Purpose.Wedding <dbl> 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ... Encoding Continuous (or Numeric) Variables
In the previous sections, we learned how to encode categorical variables. However, sometimes it may be useful to carry out encoding for numerical variables as well. For example, the Naive Bayes Algorithm requires all variables to be categorical, so encoding numerical variables is required. This is also called binning.
We will consider the Income variable as an example. Let's look at the summary statistics of this variable.
1 Min. 1st Qu. Median Mean 3rd Qu. Max. 2 133300 384975 508350 706302 766100 8444900 The values of Income range between $133,300 and $8.44 million, which shows that the distribution is right skewed. One of the additional benefits of binning is that it also takes care of the outliers. Let's create three levels of the variable Income , which are "Low" for income levels lying below $380,000, "High" for income values above $760,000, and "Mid50" for the middle 50 percentage values of the income distribution.
The first step is to create a vector of these cut-off points, which is done in the first line of code below. The second line gives the respective names to these cut-off points. The third line uses the cut() function to break the vector using the cut-off points. Finally, we compare the original Income variable with the binned Income_New variable using the summary() function.
1 bins <- c(-Inf, 384975, 766100, Inf) 2 3bin_names <- c("Low", "Mid50", "High") 4 5dat$Income_new <- cut(dat$Income, breaks = bins, labels = bin_names) 6 7summary(dat$Income) 8 9summary(dat$Income_new) {r}
1 Min. 1st Qu. Median Mean 3rd Qu. Max. 2 133300 384975 508350 706302 766100 8444900 3 4 Low Mid50 High 5 150 301 149 The above output shows that the variable has been binned. It is also possible to create bin cut-offs automatically, as shown in the code below. In this case, we create 5 bins of approximately equal width for the variable Age .
1 dat$Age_new <- cut(dat$Age, breaks = 5, labels = c("Bin1", "Bin2", "Bin3","Bin4", "Bin5")) 2 3summary(dat$Age) 4 5summary(dat$Age_new) {r}
1 Min. 1st Qu. Median Mean 3rd Qu. Max. 2 22.00 36.00 50.00 49.31 61.00 76.00 3 4Bin1 Bin2 Bin3 Bin4 Bin5 5 108 117 114 162 99 Conclusion
In this guide, you have learned methods of encoding data with R. You have applied these techniques on both quantitative and qualitative variables. Depending on the objective of your project, you can apply any or all of these encoding techniques. To learn more about data science using R, please refer to the following guides:
How to Do One Hot Encoding in R
Source: https://www.pluralsight.com/guides/encoding-data-with-r
0 Response to "How to Do One Hot Encoding in R"
Post a Comment