Skip to content

mljs/regression-robust-polynomial

Repository files navigation

regression-robust-polynomial

NPM version build status Test coverage npm download

Robust polynomial regression using LMedS.

This code is based on the implementation of this paper.

Installation

$ npm i ml-regression-robust-polynomial

Usage

import { RobustPolynomialRegression } from 'ml-regression-robust-polynomial';

var size = 30;
var x = new Array(size);
var y = new Array(size);
for (var i = 0; i < size; i++) {
  x[i] = i;
  y[i] = 2 * i * i + 4 * i + 5;
}
y[4] = y[4] * 100;
y[10] = y[10] * -100;

var regression = new RobustPolynomialRegression(x, y, 3);

regression.predict(3) === 35;

License

MIT