Often data you’re working with has abstract column names, such as (x1, x2, x3…). Typically, the first step I take when renaming columns with r is opening my web browser.
For some reason no matter the amount of times doing this it’s just one of those things. (Hoping that writing about it will change that)
The dataset cars is data from the 1920s on “Speed and Stopping Distances of Cars”. There is only 2 columns shown below.
colnames(datasets::cars)
[1] "speed" "dist"
If we wanted to rename the column “dist” to make it easier to know what the data is/means we can do so in a few different ways.
Introduction:
Employee Attrition is when an employee leaves a company due to normal means, (loss of customers, retirement, and resignation), and there is not someone to fill the vacancy. Can a company identify employee’s that are likely to leave a company?
A company with a high employee attrition rate is a good sign of underlying problems and can affect a company in a very negative way. One such way is the cost related to finding and training a replacement, as well as the possible strain it can put on other workers that in the meantime have to cover.
Preprocessing:
This dataset was produced by IBM and has just under 1500 observations of 31 different variables including attrition. 4 of the variables (EmployeeNumber, Over18, EmployeeCount, StandardHours) have the same value for all observations. Due to this, we can drop these since they won’t be helpful for our model. Next, the column “ï..Age” was renamed to “Age” to make calling this variable simpler. Finally, for build and testing models, the dataset was split into a training and test set at 70/30.
Initial Analysis:
Looking at the overall employee attrition rate for the entire dataset we can see it’s ~19%. Typically, a goal for a company is to keep this rate to ~10% and this dataset shows almost double that rate.
Here we show the influence of all factors on the employee attrition rate which shows the influence levels are similar. However, we can take the top factors and explore those in depth.
Top Factor Analysis Findings:
Factor
Variable Importance
Total Working Years
0.6564557
Years At Company
0.6525268
Overtime
0.6505954
Years In Current Role
0.6480052
Monthly Income
0.6456590
Job Level
0.6414233
Total Working Years:
Looking at the total amount of years an employee has been in the workforce (at any job) there are two significant points to be found. First, in the initial 3 years of working, the data shows the attrition rate of 50%. This is expected as people tend to start at an entry-level job and get their first job experience before moving on. The rate drops off in the following amount of years until reaching 37 – 40 years in the job force. Here we have just under ~75% attrition rate which can be best explained as employees retiring since 37 years from 18 is 55 years old, the age people usually retire at.
Years at the company:
The findings related to the number of years at the company and employee attrition followed the same trend as total working years did but with the rate lower for each. The reasoning behind this is most likely the same as total working years, with early on moving around. Then, staying put and finally retiring.
Overtime:
Employees that work overtime have over double the attrition rate (~25%), then those who don’t (~10). A possible reason behind this could be that some employees can get “burned out” working overtime. Possibly want to spend time outside of work and end up looking for a new job.
Monthly Income:
As expected employees with a higher monthly income were less likely to leave a company. Specifically, in the human resource and research and development departments. The sales department was interesting in that monthly income wasn’t as big a factor in attrition.
Model Building:
Gradient Boosting Model (GBM):
Using a GBM model with default parameters, the best training model came at 88%, at 150 trees. Using this model, we can create a prediction using the test data. The accuracy of this prediction was 87% which being very close to the training accuracy shows this is correct.
Interaction.depth
n.Trees
Accuracy
Kappa
1
150
.878
.397
Classification Trees:
The classification tree built with default parameters showed a slightly lower overall accuracy. The training accuracy came to 82% and the prediction was 83%.
dt_model<- train(Attrition ~ ., data = attrition_train, method = "rpart")
cp
Accuracy
Kappa
0.039
.82
.24
When building a classification tree with only the top 5 factors, the accuracy fell in between the other two models at, 84% training and prediction.
As we can see from this data analysis, the biggest factor to employee attrition is the length of time in the workforce either at the same company or not. However, I would recommend looking deeper into employees that work overtime and getting their reasons for leaving. Possibly, have meetings with overtime workers and find out if they need help. For example, if they are working at their capacity and still having to work overtime then might be time and possibly even cheaper to hire extra help.
I would also recommend for the company to continue to collect this same type of data at an annual basis and run the models to find those employees that are more likely to leave. Once you have the list of employees, set up reviews and see if their’s a way to help them out or even you may catch, worker issues early on. Lastly, a further review into the sales department is warranted with the high attrition rate.
Using Basic Data Analysis functions on the mtcars dataset
Let’s Start
# Copying mtcars data frame to our new data frame myCars myCars <- mtcars
Which car has the highest horsepower (hp) ?
#find and display the car with the highest horsepower index <- which.max(myCars$hp) # Display the car name along with the rest of the row myCars[index,]
## mpg cyl disp hp drat wt qsec vs am gear carb ## Maserati Bora 15 8 301 335 3.54 3.57 14.6 0 1 5 8
Maserati Bora has the highest horsepower at 335
Exploring miles per gallon (mpg) of the cars
# find and display the car with the highest mpg index<-which.max(myCars$mpg) myCars[index,]
## mpg cyl disp hp drat wt qsec vs am gear carb ## Toyota Corolla 33.9 4 71.1 65 4.22 1.835 19.9 1 1 4 1
# Creating a sorted dataframe, based on mpg highMPGcars <- myCars[ order(-myCars$mpg),] head(highMPGcars)
Learn To Manipulate Data Frames Using The “mtcars” Dataset
Task 1: Create a new column to find Displacement per Cylinder
Create a new variable (DisplacementPerCylinder), to calculate the total displacement per cylinder in cubic inches for each vehicle from the mtcars dataset.
# "str" allows you to display the internal structure of an R object
str(mtcars)
# As a backup we can copy the original data frame into a new one to work with
# That way if there is any issues we can go back
my_mtcars <- mtcars
# Calculate Displacement Per Cylinder by dividing the values (disp) and (cyl)
my_mtcars$DisplacementPerCylinder <- my_mtcars$disp / my_mtcars$cyl
# Report a summary of the variable
summary(my_mtcars$DisplacementPerCylinder)
## Min. 1st Qu. Median Mean 3rd Qu. Max.
## 17.77 26.92 34.48 35.03 43.19 59.00
Task 2: Create your own data frame
Gather data from family & friends on the number of pets they have, the birth order they are in their family and the number of siblings.
# Family/Friends ID
friendID <- c(1, 2, 3, 4, 5)
# Number of pets they have
Pets <- c(4, 4, 2, 3, 1)
# The birth order they are in their family
Order <- c(1, 2, 2, 1, 1)
# Number of Siblings
Siblings <- c(2, 2, 1, 2, 0)
# Binding the vectors into a data frame called myFriends
myFriends <- data.frame(friendID, + Pets, + Order, + Siblings)
# Command to report the structure of the data frame myFriends
str(myFriends)
## 'data.frame': 5 obs. of 4 variables:
## $ friendID : num 1 2 3 4 5
## $ X.Pets : num 4 4 2 3 1
## $ X.Order : num 1 2 2 1 1
## $ X.Siblings: num 2 2 1 2 0
# Rename the columns to get rid of the "x." in front of the names
colnames(myFriends) <- c("FriendID", "Pets", "Order", "Siblings")
str(myFriends)
## 'data.frame': 5 obs. of 4 variables:
## $ FriendID: num 1 2 3 4 5
## $ Pets : num 4 4 2 3 1
## $ Order : num 1 2 2 1 1
## $ Siblings: num 2 2 1 2 0
# Listing the values of the vector friendID from the data frame myFriends
myFriends$FriendID
## [1] 1 2 3 4 5
# Listing the values of the vector Pets from the data frame myFriends
myFriends$Pets
## [1] 4 4 2 3 1
# Listing the values of the vector Order from the data frame myFriends
myFriends$Order
## [1] 1 2 2 1 1
# Listing the values of the vector Siblings from the dataframe myFriends
myFriends$Siblings
# [1] 2 2 1 2 0
# Report a summary of the dataframe
summary(myFriends)
## FriendID Pets Order Siblings
## Min. :1 Min. :1.0 Min. :1.0 Min. :0.0
## 1st Qu.:2 1st Qu.:2.0 1st Qu.:1.0 1st Qu.:1.0
## Median :3 Median :3.0 Median :1.0 Median :2.0
## Mean :3 Mean :2.8 Mean :1.4 Mean :1.4
## 3rd Qu.:4 3rd Qu.:4.0 3rd Qu.:2.0 3rd Qu.:2.0
## Max. :5 Max. :4.0 Max. :2.0 Max. :2.0
