ml-basic
Version:
Lightweight, zero dependency, machine learning library
15 lines (14 loc) • 578 B
TypeScript
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;
}