Introducing correlationfunnel v0.1.0 - Speed Up Exploratory Data Analysis by 100X
Written by Matt Dancho
I’m pleased to announce the introduction of
correlationfunnel version 0.1.0, which officially hit CRAN yesterday. The
correlationfunnel package is a tool that enables efficient exploration of data, understanding relationships, and get to business insights as fast as possible. I’ve taught
correlationfunnel to my 500+ students enrolled in the Advanced Machine Learning course (DS4B 201-R) at Business Science University. The results have been great so far. Students are using it as the EDA step prior to Machine Learning to ensure that features have relationship before they spend significant time tuning ML models on bad data. It’s helped me get to business insights 100X faster, which has been a massive productivity boost. And, it’s a great communication tool that has helped explain business insights to executives and project stakeholders! Win-win-win.
A Critical Tool in My Machine Learning Arsenal
Why do I teach the
correlationfunnel? Why is it an 80/20 tool that I use in every analysis? Why is it so great for communication?
It’s because of the Correlation Funnel Visualization. The correlation funnel succinctly explains relationships to the target feature. Plus, every executive and business leader understands correlation. So it makes it super easy to explain results.
Here’s the correlation funnel that we’ll make in the 2-Minute Case Study - Bank Marketing Campaign. It’s examining a Marketing Campaign for a Bank (provided by the UCI Machine Learning Repository). Without knowing much about the data, we can see that the most important features (those with highest absolute correlation) go to the top. This is the crux of business insights - We know know which features are important, without spending significant time on machine learning. Further, we can make educated guesses about the story within the data (great for communicating with stakeholders).
Correlation Funnel Visualization - Important Features go to the Top
Table of Contents
1. Bank Marketing Campaign Case Study - 2 Minute Example
The following case study showcases the power of fast exploratory correlation analysis with
correlationfunnel. The goal of the analysis is to determine which features relate to the bank’s marketing campaign goal of having customers opt into a TERM DEPOSIT (financial product).
We will see that using 3 functions, we can quickly:
Transform the data into a binary format with
Perform correlation analysis using
Visualize the highest correlation features using
Rather than spend hours looking at individual plots of campaign features and comparing them to which customers opted in to the TERM DEPOSIT product, in seconds we can discover which groups of customers have enrolled, drastically speeding up EDA.
Next, load the following libraries.
Next, collect data to analyze. We’ll use Marketing Campaign Data for a Bank that was popularized by the UCI Machine Learning Repository. We can load the data with
Response & Predictor Relationships
Modeling and Machine Learning problems often involve a response (Enrolled in
TERM_DEPOSIT, yes/no) and many predictors (AGE, JOB, MARITAL, etc). Our job is to determine which predictors are related to the response. We can do this through Binary Correlation Analysis.
Binary Correlation Analysis
Binary Correlation Analysis is the process of converting continuous (numeric) and categorical (character/factor) data to binary features. We can then perform a correlation analysis to see if there is predictive value between the features and the response (target). Refer to Methodology, Key Considerations, and FAQs Documentation for more information on the methodology.
Step 1: Convert to Binary Format
The first step is converting the continuous and categorical data into binary (0/1) format. We de-select any non-predictive features. The
binarize() function then converts the features into binary features.
Numeric Features: Are binned into ranges or if few unique levels are binned by their value, and then converted to binary features via one-hot encoding
Categorical Features: Are binned by one-hot encoding
The result is a data frame that has only binary data with columns representing the bins that the observations fall into. There are now 74 columns that are binary (0/1). Note that the output was trimmed to the first 10 rows using
Step 2: Perform Correlation Analysis
The second step is to perform a correlation analysis between the response (
target = TERM_DEPOSIT_yes) and the rest of the features. This returns a specially formatted tibble with the feature, the bin, and the bin’s correlation to the target. The format is exactly what we need for the next step - Producing the Correlation Funnel
Step 3: Visualize the Correlation Funnel
A Correlation Funnel is an tornado plot that lists the highest correlation features (based on absolute magnitude) at the top of the and the lowest correlation features at the bottom. The resulting visualization looks like a Funnel.
To produce the Correlation Funnel, use
plot_correlation_funnel(). Try setting
interactive = TRUE to get an interactive plot that can be zoomed in on.
Examining the Results
The most important features are towards the top. We can investigate these.
We can see that the following prospect groups have a much greater correlation with enrollment in the TERM DEPOSIT product:
When the “DURATION”, the amount of time a prospect is engaged in marketing campaign material, is 319 seconds or longer.
When “POUTCOME”, whether or not a prospect has previously enrolled in a product, is “success”.
When “CONTACT”, the medium used to contact the person, is “cellular”
When “HOUSING”, whether or not the contact has a HOME LOAN is “no”
2. My EDA Workflow R
The main addition of
correlationfunnelto the EDA ecosystem in
Ris to quickly expose feature relationships.
There are other EDA packages, but none implement the binarization process as a preprocessing step before correlation. This is critical to extracting insights especially in nonlinear continuous data.
Correlation funnel requires semi-processed data meaning
- Missing (
NA) values have been treated
- Date or date-time features have been feature engineered into categorical variables
- Data is in a “clean” format (numeric data and categorical data are ready to be correlated to a Yes/No response)
Because of this, I use other packages to understand data (e.g. missing values) and correct data (preprocessing).
EDA Packages I Use Before I Use Correlation Funnel to identify processing steps
GGally - The
ggpairs()function is one of my all-time favorites for visualizing many features quickly. I teach
GGallyin DS4B 201-R, Advanced Machine Learning and Business Consulting, part of the 3-Course R-Track for Business.
skimr - The
skim()function is my go-to for console statistics on each of my variables. It’s like
summary()on steroids. I teach
skimrin DS4B 201-R, Advanced Machine Learning and Business Consulting, part of the 3-Course R-Track for Business.
Data Explorer - Automates Exploration and Data Treatment. Amazing for investigating features quickly and efficiently including by data type, missing data, feature engineering, and identifying relationships.
Packages I use to implement preprocessing
- recipes - This is one of the most important R packages in my exploratory and machine learning workflow. It’s use to create preprocessing pipelines - A series of steps to correct data and prepare for machine learning, correlation analysis, etc. I teach
recipesin DS4B 201-R, Advanced Machine Learning and Business Consulting, part of the 3-Course R-Track for Business.
Other Great EDA Packages (Notable Mentions)
VisDat - Similar in concept to Data Explorer. Utilities for visualizing missing data
naniar - For understanding missing data.
UpSetR - For generating upset plots
3. What do you think of Correlation Funnel?
Let me know what to do you think of
correlationfunnel in the comments. I’m interested in getting your feedback on how it works on different data sets. I’ve tried it on a ton, but that doesn’t mean that it’s perfect. Let me know your results - good or bad.
4. Help Others Learn R - Share this article if you like it!
Post the article on LinkedIn. Tag Matt Dancho (@Matt Dancho), and I’ll respond!
5. New Expert Shiny Course - Coming Soon!
Shiny Apps! Build these two apps: