CRAN status

DriveML

DriveML for automated machine learning. DriveML helps in implementing some of the pillars of an automated machine learning pipeline such as automated data preparation, feature engineering, model building, and model explanation by executing simple functions for each purpose instead of writing lengthy R codes.

DriveML Framework:

DriveML

DriveML is a series of functions sucha as AutoDataPrep, AutoMAR, autoMLmodel. DriveML automates some of the most difficult machine learning functions such as data cleaning, data tranformations, feature engineering, model training, model validation, model tuning and model selection.

Functionalities of DriveML

Three key features of DriveML : Pre-processing, ML Techniques and Model interpretations

  1. Automatic data preparation
    • AutoDataPrep function to generate a novel features based on the functional understanding of the dataset
  2. ML Model Optimization
    • autoMLmodel function to develope baseline machine learning models using regression and tree based classfication techniques
  3. Model interpretations
    • autoMLReport function to print the machine learning model outcome in HTML format

Installation

Install from CRAN within R using:

install.packages("DriveML")

Install the latest development verion of the DriveML from github with:

devtools::install_github("daya6489/DriveML")

Example

Data

In this vignette, we will be using Heart Disease - Classifications data set

Data source UCI

    library("DriveML")
    library("SmartEDA")
    ## Load heart disease dataset 
    data(heart)

Overview of the data

Understanding the dimensions of the dataset, variable names, overall missing summary and data types of each variables

## overview of the data; 
    ExpData(data = heart, type = 1)
## structure of the data    
    ExpData(data = heart, type = 2)

Summary of numerical variables

To summarise the numeric variables, you can use following r codes from this pacakge

## Summary statistics by – overall
    ExpNumStat(heart, by = "GA", gp = "target_var", Qnt = seq(0, 1, 0.1), MesofShape = 2, Outlier = TRUE, round = 2)

Graphical representation of all numeric features

## Generate Boxplot by category
ExpNumViz(heart, gp = "target_var", type = 2, nlim = 25, Page = c(2, 2))

## Generate Density plot
ExpNumViz(heart, gp = NULL, type = 3, nlim = 10, Page = c(2, 2))

## Generate Scatter plot
ExpNumViz(heart, target="target_var", nlim = 4, scatter = TRUE, Page=c(2, 1))

Machine learning model using 3 simple steps

Data preparation using autoDataprep function

Single function to prepare end to end data preparation for machine learning models

# Data Preparation
small_data <- autoDataprep(heart, target = "target_var", missimpute = "default",
                       auto_mar = TRUE, mar_object = NULL, dummyvar = TRUE,
                       char_var_limit = 12, aucv = 0.02, corr = 0.99,
                       outlier_flag = TRUE, interaction_var = TRUE,
                       frequent_var = TRUE, uid = NULL, onlykeep = NULL, drop = NULL)

# Print output on R console
printautoDataprep(small_data)

# Final prepared master data

small_data_t <- small_data$master_data

Machine learning classification model using autoMLmodel function

Single function to develope six different types of machine learning binary classification models with the help of hyperparameter tuining using random search

# DriveML Model development
small_ml_random <- autoMLmodel(small_data_t,  target = "target_var",  testSplit = 0.2,
                      tuneIters = 5,  tuneType = "random",
                      models = "all", varImp = 10,  liftGroup = 10,  maxObs = 10000,  uid = NULL,
                      pdp = T, positive = 1, htmlreport = FALSE, seed = 1991)

# Model summary results
small_ml_random$results

Present model report using autoMLReport function

Generate a report in html format for the output of autoDataprep and autoMLmodel DriveML fucntions. Alos autoMLReport is an inbuilt function under autoMLmodel.

autoMLReport(mlobject = small_ml_random, mldata = small_data, op_file = "driveML_ouput_heart_data.html")

Articles

The pre-print version of the paper on DriveML is available at ArXiv at- https://arxiv.org/pdf/2005.00478.pdf. Also, currently the paper is under review at the Journal of Open Source Software (JOSS).