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

Lightweight, zero dependency, machine learning library

import Matrix from "../lib/matrix";
import Optimizer from "./optimizer";
export type GradientDescentParams = {
    /**
     * @default 0.01
     */
    learningRate?: number;
    /**
     * @default 0
     */
    clipping?: number;
};
export default class GradientDescent extends Optimizer {
    name: string;
    learningRate: number;
    clipping: number;
    constructor({ learningRate, clipping }?: GradientDescentParams);
    step(input: Matrix, gradient: Matrix, callback: (input: Matrix, gradient: Matrix) => void): void;
}