Classes

Methods

static
accuracyScore(yTrue, yPred) → number

Accuracy score is the proportion of true results (both true positives and true negatives) among the total number of cases examined. Formula: (tp + tn) / (tp + fp + fn + tn).

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Predicted (estimated) lables.

Returns

number Accuracy value.

static
bestF1Threshold(yTrue, yPred) → number

Gets threshold for prediction score, which results in the highest F1.

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Estimated probabilities.

Returns

number Threshold with highest F1 score.

static
breakEventPointScore(yTrue, yPred) → number

Get break-even point, the value where precision and recall intersect.

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Estimated probabilities.

Returns

number Break-even point score.

static
desiredPrecisionThreshold(yTrue, yPred, desiredPrecision) → number

Gets threshold for prediction score, nearest to specified precision.

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Estimated probabilities.

desiredPrecision

number

 

Desired precision score.

Returns

number Threshold for prediction score, nearest to specified precision.

static
desiredRecallThreshold(yTrue, yPred, desiredRecall) → number

Gets threshold for recall score, nearest to specified recall.

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Estimated probabilities.

desiredRecall

number

 

Desired recall score.

Returns

number Threshold for recall score, nearest to specified recall.

static
f1Score(yTrue, yPred) → number

The F1 score can be interpreted as a weighted average of the precision and recall, where an F1 score reaches its best value at 1 and worst score at 0. The relative contribution of precision and recall to the F1 score are equal. Formula: 2 * (precision * recall) / (precision + recall).

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Predicted (estimated) lables.

Returns

number F1 score.

static
meanAbsoluteError(yTrueVec, yPredVec) → number

Mean absolute error (MAE) regression loss.

Parameters

Name Type Optional Description

yTrueVec

(Array of number or module:la.Vector)

 

ground truth values in yTrueVec.

yPredVec

(Array of number or module:la.Vector)

 

estimated values in yPredVec.

Returns

number Error value.

static
meanAbsolutePercentageError(yTrueVec, yPredVec) → number

Mean absolute percentage error (MAPE) regression loss.

Parameters

Name Type Optional Description

yTrueVec

(Array of number or module:la.Vector)

 

ground truth values in yTrueVec.

yPredVec

(Array of number or module:la.Vector)

 

estimated values in yPredVec.

Returns

number Error value.

static
meanError(yTrueVec, yPredVec) → number

Mean error (ME) regression loss.

Parameters

Name Type Optional Description

yTrueVec

(Array of number or module:la.Vector)

 

ground truth values in yTrueVec.

yPredVec

(Array of number or module:la.Vector)

 

estimated values in yPredVec.

Returns

number Error value.

static
meanSquareError(yTrueVec, yPredVec) → number

Mean square error (MSE) regression loss.

Parameters

Name Type Optional Description

yTrueVec

(Array of number or module:la.Vector)

 

ground truth values in yTrueVec.

yPredVec

(Array of number or module:la.Vector)

 

estimated values in yPredVec.

Returns

number Error value.

static
precisionRecallCurve(yTrue, yPred[, sample]) → module:la.Matrix

Get precision recall curve sampled on sample points.

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Estimated probabilities.

sample

number

Yes

Desired number of samples in output.

Defaults to 10.

Returns

module:la.Matrix Precision-recall pairs.

static
precisionScore(yTrue, yPred) → number

Precision score is defined as the proportion of the true positives against all the positive results (both true positives and false positives). Formula: tp / (tp + fp).

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Predicted (estimated) lables.

Returns

number Precission score.

static
r2Score(yTrueVec, yPredVec) → number

R^2 (coefficient of determination) regression score.

Parameters

Name Type Optional Description

yTrueVec

(Array of number or module:la.Vector)

 

ground truth values in yTrueVec.

yPredVec

(Array of number or module:la.Vector)

 

estimated values in yPredVec.

Returns

number Error value.

static
recallScore(yTrue, yPred) → number

Recall score is intuitively the ability of the classifier to find all the positive samples. Formula: tp / (tp + fn).

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Predicted (estimated) lables.

Returns

number Recall score.

static
rocAucScore(yTrue, yPred[, sample]) → number

Get AUC of the current curve.

Example

// import metrics module
var metrics = require('qminer').analytics.metrics;

// true and predicted lables
var true_lables = [0, 1, 0, 0, 1];
var pred_prob = [0.3, 0.5, 0.2, 0.5, 0.8];

// compute ROC curve
var auc = metrics.rocAucScore(true_lables, pred_prob); // output: 0.92

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Estimated probabilities.

sample

number

Yes

Desired number of samples in output.

Defaults to 10.

Returns

number Area under ROC curve.

static
rocCurve(yTrue, yPred[, sample]) → module:la.Matrix

Get ROC parametrization sampled on sample points.

Example

// import metrics module
var metrics = require('qminer').analytics.metrics;

// true and predicted lables
var true_lables = [0, 1, 0, 0, 1];
var pred_prob = [0.3, 0.5, 0.2, 0.5, 0.8];

// compute ROC curve
var roc = metrics.rocCurve(true_lables, pred_prob); // output: [ [ 0, 0 ], [0, 0.5], [[ 0.34, 1 ],], [ 0.67, 0 ], [ 1, 1 ] ]

Parameters

Name Type Optional Description

yTrue

(Array of number or module:la.Vector)

 

Ground truth (correct) lables.

yPred

(Array of number or module:la.Vector)

 

Estimated probabilities.

sample

number

Yes

Desired number of samples in output.

Defaults to 10.

Returns

module:la.Matrix A matrix with increasing false and true positive rates.

static
rootMeanSquareError(yTrueVec, yPredVec) → number

Root mean square (RMSE) error regression loss.

Parameters

Name Type Optional Description

yTrueVec

(Array of number or module:la.Vector)

 

ground truth values in yTrueVec.

yPredVec

(Array of number or module:la.Vector)

 

estimated values in yPredVec.

Returns

number Error value.