UNPKG

ml-basic

Version:

Lightweight, zero dependency, machine learning library

15 lines (14 loc) 578 B
import { Layers } from "../layers"; import Layer from "../layers/layer"; import Optimizer, { HyperParameters } from "../optimizers/optimizer"; import { LossFunction } from "./functions"; import Matrix from "./matrix"; export default class Network { type: string; layers: Layer[]; lossFunction: LossFunction; constructor(layers: Layers[], optimizer: Optimizer, lossFunction: LossFunction); propagate(input: Matrix, disableDropout?: boolean): Matrix[]; backPropagate(input: Matrix, target: Matrix): number; configure(options: HyperParameters): void; }