toynn v2.0.5

Requirements

NodeJS v18.0.0 or higher must be installed to use toynn.

Installation

Usage

import toynn from "toynn";

const X = [
  new toynn.NArray([0, 0]).reshape(1, 2),
  new toynn.NArray([0, 1]).reshape(1, 2),
  new toynn.NArray([1, 0]).reshape(1, 2),
  new toynn.NArray([1, 1]).reshape(1, 2),
];

const y = [
  new toynn.NArray([0]),
  new toynn.NArray([0]),
  new toynn.NArray([0]),
  new toynn.NArray([1]),
];

const model = new toynn.NN("and");

const layer1 = new toynn.Layer(2, 3);
layer1.use(toynn.functions.linear);
const layer2 = new toynn.Layer(3, 1);
layer2.use(toynn.functions.sigmoid);

model.add(layer1);
model.add(layer2);

model.train({
  x: X,
  y,
  epochs: 500,
  alpha: 0.001,
  loss: toynn.errors.MSE,
  verbose: true,
});

let newData = new toynn.NArray([1, 0]).reshape(1, 2);
// make prediction
console.log(model.forward(newData).flatten());

Note: The above code only supports v2.0.0 or above

You can use the library with typescript also. The code remains the same.

Docs

Docs can be found here.