RcppMLPACK2 and the MLPACK Machine Learning Library

Dirk Eddelbuettel — written Feb 20, 2017 — source

mlpack

mlpack is, to quote, a scalable machine learning library, written in C++, that aims to provide fast, extensible implementations of cutting-edge machine learning algorithms. It has been written by Ryan Curtin and others, and is described in two papers in BigLearning (2011) and JMLR (2013). mlpack uses Armadillo as the underlying linear algebra library, which, thanks to RcppArmadillo, is already a rather well-known library in the R ecosystem.

RcppMLPACK1

Qiang Kou has created the RcppMLPACK package on CRAN for easy-to-use integration of mlpack with R. It integrates the mlpack sources, and is, as a CRAN package, widely available on all platforms.

However, this RcppMLPACK package is also based on a by-now dated version of mlpack. Quoting again: mlpack provides these algorithms as simple command-line programs and C++ classes which can then be integrated into larger-scale machine learning solutions. Version 2 of the mlpack sources switched to a slightly more encompassing build also requiring the Boost libraries ‘program_options’, ‘unit_test_framework’ and ‘serialization’. Within the context of an R package, we could condition out the first two as R provides both the direct interface (hence no need to parse command-line options) and also the testing framework. However, it would be both difficult and potentially undesirable to condition out the serialization which allows mlpack to store and resume machine learning tasks.

We refer to this version now as RcppMLPACK1.

RcppMLPACK2

As of February 2017, the current version of mlpack is 2.1.1. As it requires external linking with (some) Boost libraries as well as with Armadillo, we have created a new package RcppMLPACK2 inside a new GitHub organization RcppMLPACK.

Linux

This package works fine on Linux provided mlpack, Armadillo and Boost are installed.

OS X / macOS

For maxOS / OS X, James Balamuta has tried to set up a homebrew recipe but there are some tricky interaction with the compiler suites used by both brew and R on macOS.

Windows

For Windows, one could do what Jeroen Ooms has done and build (external) libraries. Volunteers are encouraged to get in touch via the issue tickets at GitHub.

Installation from source

Release are available from a drat repository hosted in the GitHub orgranization RcppMLPACK. So

drat:::add("RcppMLPACK")         # first add the repo
install.package("RcppMLPACK2")   # install the pacage
update.packages()                # or update to newer one (if one exists)

will use this. If you prefer to rather pick a random commit state,

remotes::install_github("rcppmlpack/rcppmlpack2")

will work as well.

Example: Logistic Regression

To illustrate mlpack we show a first simple example also included in the package. As the rest of the Rcpp Gallery, these are “live” code examples.

#include <RcppMLPACK.h>				// MLPACK, Rcpp and RcppArmadillo

#include <mlpack/methods/logistic_regression/logistic_regression.hpp> 	// particular algorithm used here

// [[Rcpp::depends(RcppMLPACK)]]

// [[Rcpp::export]]
Rcpp::List logisticRegression(const arma::mat& train,
                              const arma::irowvec& labels,
                              const Rcpp::Nullable<Rcpp::NumericMatrix>& test = R_NilValue) {
    
    // MLPACK wants Row<size_t> which is an unsigned representation
    // that R does not have
    arma::Row<size_t> labelsur, resultsur;

    // TODO: check that all values are non-negative
    labelsur = arma::conv_to<arma::Row<size_t>>::from(labels);

    // Initialize with the default arguments.
    // TODO: support more arguments>
    mlpack::regression::LogisticRegression<> lrc(train, labelsur);
    
    arma::vec parameters = lrc.Parameters();

    Rcpp::List return_val;
    
    if (test.isNotNull()) {
        arma::mat test2 = Rcpp::as<arma::mat>(test);
        lrc.Classify(test2, resultsur);
        arma::vec results = arma::conv_to<arma::vec>::from(resultsur);
        return_val = Rcpp::List::create(Rcpp::Named("parameters") = parameters,
                                        Rcpp::Named("results") = results);
    } else {
        return_val = Rcpp::List::create(Rcpp::Named("parameters") = parameters);
    }

    return return_val;

}

We can then call this function with the same (trivial) data set as used in the first unit test for it:

logisticRegression(matrix(c(1, 2, 3, 1, 2, 3), nrow=2, byrow=TRUE), c(1L, 1L, 0L))
$parameters
[1]  67.9550 -13.6328 -13.6328

Example: Naive Bayes Classifier

A second examples shows the NaiveBayesClassifier class.

#include <RcppMLPACK.h>				// MLPACK, Rcpp and RcppArmadillo

#include <mlpack/methods/naive_bayes/naive_bayes_classifier.hpp> 	// particular algorithm used here

// [[Rcpp::depends(RcppMLPACK)]]

// [[Rcpp::export]]
Rcpp::List naiveBayesClassifier(const arma::mat& train,
                                const arma::irowvec& labels,
                                const int& classes,
                                const Rcpp::Nullable<Rcpp::NumericMatrix>& test = R_NilValue) {

    // MLPACK wants Row<size_t> which is an unsigned representation
    // that R does not have
    arma::Row<size_t> labelsur, resultsur;

    // TODO: check that all values are non-negative
    labelsur = arma::conv_to<arma::Row<size_t>>::from(labels);

    // Initialize with the default arguments.
    // TODO: support more arguments>
    mlpack::naive_bayes::NaiveBayesClassifier<> nbc(train, labelsur, classes);

    Rcpp::List return_val;
    if (test.isNotNull()) {
        arma::mat armatest = Rcpp::as<arma::mat>(test);
        nbc.Classify(armatest, resultsur);
    
        arma::irowvec results = arma::conv_to<arma::irowvec>::from(resultsur);
        return Rcpp::List::create(Rcpp::Named("means") = nbc.Means(),
                                  Rcpp::Named("variances") = nbc.Variances(),
                                  Rcpp::Named("probabilities") = nbc.Probabilities(),
                                  Rcpp::Named("classification") = results);
    } else {
        return Rcpp::List::create(Rcpp::Named("means") = nbc.Means(),
                                  Rcpp::Named("variances") = nbc.Variances(),
                                  Rcpp::Named("probabilities") = nbc.Probabilities());
    }
}

We can use the sample data included in recent-enough version of the RcppMLPACK package:

library(RcppMLPACK)
data(trainSet)                ## data part of RcppMLPACK package (when using RcppMLPACK2 source)
trainmat <- t(trainSet[, -5]) ## train data
trainlab <- trainSet[, 5]     ## labels
naiveBayesClassifier(trainmat, trainlab, 2L)                   ## just model
$means
[1] 2.75000 4.00000 3.68750 2.37500 8.33333 4.66667 3.66667 2.40000

$variances
[1] 0.333333 0.800000 0.629167 0.383333 0.809524 3.380952 0.666667 0.400000

$probabilities
[1] 0.516129 0.483871
testmat <- t(testSet[, -5])   ## test data
testlab <- testSet[, 5]             
res <- naiveBayesClassifier(trainmat, trainlab, 2L, testmat)   ## also classify
res
$means
[1] 2.75000 4.00000 3.68750 2.37500 8.33333 4.66667 3.66667 2.40000

$variances
[1] 0.333333 0.800000 0.629167 0.383333 0.809524 3.380952 0.666667 0.400000

$probabilities
[1] 0.516129 0.483871

$classification
[1] 0 0 0 1 1 1 1
## res was a rowvector but comes back as 1-row matrix
all.equal(res[[4]],  testlab)
[1] TRUE

As we can see, the computed classification on the test set corresponds to the expected classification in testlabels.

tags: machine_learning  armadillo  mlpack 

Related Articles