simple-ml-regression

A simple library for training a polynomial regression model from csv training data

Usage no npm install needed!

<script type="module">
  import simpleMlRegression from 'https://cdn.skypack.dev/simple-ml-regression';
</script>

README

Simple ML Regression

npm npm bundle size CI

A simple library for training a polynomial regression model from csv training data.

It uses an iterative training algorithm to optimise the model parameters and will return a model that has been trained on all available data.

Install

npm install simple-ml-regression

Usage

Given csv training data for the function x1^2 + 4x2 + x3:

X1,X2,X3,Y1
1,1,1,6
2,3,4,20
2,1,1,8
0,1,0,3
etc...

Train a polynomial model based on this training data:

import { loadDataCSV, train, predict } from 'simple-ml-regression';

loadDataCSV('data.csv', ['X1', 'X2', 'X3'], ['Y1']).then((data) => {
    const model = train(data, {
        testSplit: 0.2,
        maxDegree: 12,
        epochs: 50
    });

    const predictedResult = predict(model, [2, 5, 4]);

    console.log(predictedResult);
});

Result:

[23]