Implementation Examples

Neural Network Graph Configuration

import { NeuralNetwork, ActivationEnum, MSELoss } from 'tensorforgejs';

const network = new NeuralNetwork(
  { size: 4 },
  [
    { size: 8, activation: ActivationEnum.ReLU },
    { size: 4, activation: ActivationEnum.ReLU }
  ],
  { size: 2, activation: ActivationEnum.Softmax },
  new MSELoss()
);

const output = network.forward([0.5, 0.8, 0.1, 0.3]);
console.log('Result:', output.toArray());

K-Nearest Neighbors

import KNN from '../src/models/knn/KNN';

const knn = new KNN([
  { data: [1, 2], label: 'Class A' },
  { data: [8, 9], label: 'Class B' }
], 3);

console.log(knn.predict([1.5, 2.5]));