ml-basic/dist/types/optimizers/batch-gradient-descent.d.ts

Lightweight, zero dependency, machine learning library

import Matrix from "../lib/matrix";
import GradientDescent from "./gradient-descent";
export type BatchGradientDescentParams = {
    /**
     * @default 0.01
     */
    learningRate?: number;
    /**
     * @default 0
     */
    clipping?: number;
    /**
     * @default 8
     */
    batchSize?: number;
};
export default class BatchGradientDescent extends GradientDescent {
    name: string;
    i: number;
    batchSize: number;
    private batch?;
    constructor({ learningRate, clipping, batchSize }?: BatchGradientDescentParams);
    step(input: Matrix, gradient: Matrix, callback: (input: Matrix, gradient: Matrix) => void): void;
}