ml-basic/dist/types/optimizers/optimizer.d.ts

Lightweight, zero dependency, machine learning library

import Matrix from "../lib/matrix";
export type HyperParameters = {
    learningRate?: number;
    clipping?: number;
    batchSize?: number;
    beta1?: number;
    beta2?: number;
    epsilon?: number;
};
export default abstract class Optimizer {
    type: string;
    abstract name: string;
    abstract step(input: Matrix, gradient: Matrix, callback: (input: Matrix, gradient: Matrix) => void): void;
    clone(): any;
    configure(options: HyperParameters): void;
}