2.0.4 • Published 2 years ago

ml-regression-multivariate-linear v2.0.4

Weekly downloads
5,073
License
MIT
Repository
github
Last release
2 years ago

ml-regression-multivariate-linear

NPM version build status npm download

Multivariate linear regression.

Installation

npm install --save ml-regression-multivariate-linear

API

new MLR(x, y, options)

Arguments

  • x: Matrix containing the inputs
  • y: Matrix containing the outputs

Options

  • intercept: boolean indicating if intercept terms should be computed (default: true)
  • statistics: boolean for calculating and returning regression statistics (default: true)

Usage

import MLR from "ml-regression-multivariate-linear";

const x = [
  [0, 0],
  [1, 2],
  [2, 3],
  [3, 4]
];
// Y0 = X0 * 2, Y1 = X1 * 2, Y2 = X0 + X1
const y = [
  [0, 0, 0],
  [2, 4, 3],
  [4, 6, 5],
  [6, 8, 7]
];
const mlr = new MLR(x, y);
console.log(mlr.predict([3, 3]));
// [6, 6, 6]

License

MIT