incrBinaryClassification

Usage

var incrBinaryClassification = require( '@stdlib/ml/incr/binary-classification' );

incrBinaryClassification( N[, options] )

Returns an accumulator function which incrementally performs binary classification using stochastic gradient descent.

// Create an accumulator for performing binary classification on 3-dimensional data:
var accumulator = incrBinaryClassification( 3 );

The function accepts the following options:

• intercept: boolean indicating whether to include an intercept. If true, an element equal to one is implicitly added to each provided feature vector (note, however, that the model does not perform regularization of the intercept term). If false, the model assumes that feature vectors are already centered. Default: true.

• lambda: regularization parameter. The regularization parameter determines the amount of shrinkage inflicted on the model coefficients. Higher values reduce the variance of the model coefficient estimates at the expense of introducing bias. Default: 1.0e-4.

• learningRate: an array-like object containing the learning rate function and associated parameters. The learning rate function decides how fast or slow the model coefficients will be updated toward the optimal coefficients. Must be one of the following:

  • ['constant', ...]: constant learning rate function. To set the learning rate, provide a second array element. By default, when the learn rate function is 'constant', the learning rate is set to 0.02.
  • ['basic']: basic learning rate function according to the formula 10/(10+t) where t is the current iteration.
  • ['invscaling', ...]: inverse scaling learning rate function according to the formula eta0/pow(t, power_t) where eta0 is the initial learning rate and power_t is the exponent controlling how quickly the learning rate decreases. To set the initial learning rate, provide a second array element. By default, the initial learning rate is 0.02. To set the exponent, provide a third array element. By default, the exponent is 0.5.
  • ['pegasos']: Pegasos learning rate function according to the formula 1/(lambda*t) where t is the current iteration and lambda is the regularization parameter.

  Default: ['basic'].

• loss: loss function. Must be one of the following:

  • hinge: hinge loss function. Corresponds to a soft-margin linear Support Vector Machine (SVM), which can handle non-linearly separable data.
  • log: logistic loss function. Corresponds to Logistic Regression.
  • modifiedHuber: Huber loss function variant for classification.
  • perceptron: hinge loss function without a margin. Corresponds to the original perceptron by Rosenblatt (1957).
  • squaredHinge: squared hinge loss function SVM (L2-SVM).

  Default: 'log'.

By default, the model contains an intercept term. To omit the intercept, set the intercept option to false:

var array = require( '@stdlib/ndarray/array' );

// Create a model with the intercept term:
var acc = incrBinaryClassification( 2, {
    'intercept': true
});
var coefs = acc( array( [ 1.4, 0.5 ] ), 1 );
// returns <ndarray>

var dim = coefs.length;
// returns 3

// Create a model without the intercept term:
acc = incrBinaryClassification( 2, {
    'intercept': false
});
coefs = acc( array( [ 1.4, 0.5 ] ), -1 );
// returns <ndarray>

dim = coefs.length;
// returns 2

accumulator( x, y )

If provided a feature vector x and response value y (either +1 or -1), the accumulator function updates a binary classification model; otherwise, the accumulator function returns the current binary classification model coefficients.

var array = require( '@stdlib/ndarray/array' );

// Create an accumulator:
var acc = incrBinaryClassification( 2 );

// Provide data to the accumulator...
var x = array( [ 1.0, 0.0 ] );

var coefs = acc( x, -1 );
// returns <ndarray>

x.set( 0, 0.0 );
x.set( 1, 1.0 );

coefs = acc( x, 1 );
// returns <ndarray>

x.set( 0, 0.5 );
x.set( 1, 1.0 );

coefs = acc( x, 1 );
// returns <ndarray>

coefs = acc();
// returns <ndarray>

accumulator.predict( X[, type] )

Computes predicted response values for one or more observation vectors X.

var array = require( '@stdlib/ndarray/array' );

// Create a model with the intercept term:
var acc = incrBinaryClassification( 2 );

// ...

var label = acc.predict( array( [ 0.5, 2.0 ] ) );
// returns <ndarray>

Provided an ndarray having shape (..., N), where N is the number of features, the returned ndarray has shape (...) (i.e., the number of dimensions is reduced by one) and data type float64. For example, if provided a one-dimensional ndarray, the method returns a zero-dimensional ndarray whose only element is the predicted response value.

By default, the method returns the predict label (type='label'). In order to return a prediction probability of a +1 response value given either the logistic (log) or modified Huber (modifiedHuber) loss functions, set the second argument to 'probability'.

var array = require( '@stdlib/ndarray/array' );

// Create a model with the intercept term:
var acc = incrBinaryClassification( 2, {
    'loss': 'log'
});

// ...

var phat = acc.predict( array( [ 0.5, 2.0 ] ), 'probability' );
// returns <ndarray>

In order to return the linear predictor (i.e., the signed distance to the hyperplane, which is computed as the dot product between the model coefficients and the provided feature vector x, plus the intercept), set the second argument to 'linear'.

var array = require( '@stdlib/ndarray/array' );

// Create a model with the intercept term:
var acc = incrBinaryClassification( 2, {
    'loss': 'log'
});

// ...

var lp = acc.predict( array( [ 0.5, 2.0 ] ), 'linear' );
// returns <ndarray>

Given a feature vector x = [x_0, x_1, ...] and model coefficients c = [c_0, c_1, ...], the linear predictor is equal to (x_0*c_0) + (x_1*c_1) + ... + c_intercept.