NAME

AI::XGBoost::CAPI - Perl wrapper for XGBoost C API https://github.com/dmlc/xgboost

VERSION

version 0.1

SYNOPSIS

use 5.010;
use AI::XGBoost::CAPI qw(:all);

my $dtrain = XGDMatrixCreateFromFile('agaricus.txt.train');
my $dtest = XGDMatrixCreateFromFile('agaricus.txt.test');

my ($rows, $cols) = (XGDMatrixNumRow($dtrain), XGDMatrixNumCol($dtrain));
say "Train dimensions: $rows, $cols";

my $booster = XGBoosterCreate([$dtrain]);

for my $iter (0 .. 10) {
    XGBoosterUpdateOneIter($booster, $iter, $dtrain);
}

my $predictions = XGBoosterPredict($booster, $dtest);
# say join "\n", @$predictions;

XGBoosterFree($booster);
XGDMatrixFree($dtrain);
XGDMatrixFree($dtest);

DESCRIPTION

Perlified wrapper for the C API

Error handling

XGBoost c api functions returns some int to signal the presence/absence of error. In this module that is achieved using Exceptions from Exception::Class

FUNCTIONS

XGDMatrixCreateFromFile

Load a data matrix

Parameters:

filename

the name of the file

silent

whether print messages during loading

Returns a loaded data matrix

XGDMatrixCreateFromMat

Create from dense matrix

Parameters:

matrix

matrix data

missing

value indicating missing data (optional)

Returns a loaded data matrix

XGDMatrixNumRow

Get number of rows

Parameters:

matrix

DMatrix

XGDMatrixNumCol

Get number of cols

Parameters:

matrix

DMatrix

XGDMatrixSetFloatInfo

XGDMatrixGetFloatInfo

XGDMatrixSetUintInfo

XGDMatrixGetUintInfo

XGDMatrixSaveBinary

XGDMatrixSliceDMatrix

XGDMatrixFree

Free space in data matrix

Parameters:

matrix

DMatrix to be freed

XGBoosterCreate

Create XGBoost learner

Parameters:

matrices

matrices that are set to be cached

XGBoosterSetParam

XGBoosterSetAttr

XGBoosterGetAttr

XGBoosterGetAttrNames

XGBoosterUpdateOneIter

Update the model in one round using train matrix

Parameters:

booster

XGBoost learner to train

iter

current iteration rounds

train_matrix

training data

XGBoosterBoostOneIter

XGBoosterEvalOneIter

XGBoosterPredict

Make prediction based on train matrix

Parameters:

booster

XGBoost learner

data_matrix

Data matrix with the elements to predict

option_mask

bit-mask of options taken in prediction, possible values

  • 0: normal prediction

  • 1: output margin instead of transformed value

  • 2: output leaf index of trees instead of leaf value, note leaf index is unique per tree

  • 4: output feature contributions to individual predictions

ntree_limit

limit number of trees used for prediction, this is only valid for boosted trees when the parameter is set to 0, we will use all the trees

Returns an arrayref with the predictions corresponding to the rows of data matrix

XGBoosterDumpModel

XGBoosterDumpModelEx

XGBoosterDumpModelWithFeatures

XGBoosterDumpModelExWithFeatures

XGBoosterFree

Free booster object

Parameters:

booster

booster to be freed

AUTHOR

Pablo Rodríguez González <pablo.rodriguez.gonzalez@gmail.com>

COPYRIGHT AND LICENSE

Copyright (c) 2017 by Pablo Rodríguez González.