imgcook / datacook

Machine Learning and Data Analysis in JavaScript.

DataCook

Machine learning and data science library for Javascript / Typescript.

---

Getting started

Dependencies

DataCook is built for javascript environment and can run in both node.js platform and browser. DataCook relies on tensorflow.js for basic numeric computation.

Quick installation

DataCook can be installed by npm:

npm install @pipcook/datacook

or by yarn

yarn add @pipcook/datacook

Quick start: Train a simple linear-regression model

import { Model } from '@pipcook/datacook';

const { LinearRegression } = Model;

const X = [
  [4, 5],
  [2, 3],
  [1, 4],
  [3, 8],
];
const y = [10, 5.5, 6.5, 12];
// create model
const lm = new LinearRegression();
// train linear model using feature set X and label set y
await lm.fit(X, y);
// get prediction
const yPred = lm.predict(X);
yPred.print();
// [10, 6, 6, 12]