UNPKG

ml-basic

Version:

Lightweight, zero dependency, machine learning library

20 lines (19 loc) 581 B
import { Activator } from "../lib/functions"; import Matrix from "../lib/matrix"; import Layer from "./layer"; export type FullyConnectedParams = { input: number | [number, number]; output: number; /** * @default {@link Sigmoid} */ activation?: Activator; }; export default class FullyConnectedLayer extends Layer { name: string; weights: Matrix; bias: Matrix; constructor({ input, output, activation }: FullyConnectedParams); propagate(input: Matrix): Matrix; backPropagate(input: Matrix, output: Matrix, loss: Matrix): Matrix; }