ml-basic/dist/types/layers/dropout.d.ts

Lightweight, zero dependency, machine learning library

import Matrix from "../lib/matrix";
import Layer from "./layer";
export type DropoutParams = {
    input: number | [number, number];
    /**
     * @default 0.25
     */
    rate?: number;
};
export default class DropoutLayer extends Layer {
    name: string;
    rate: number;
    constructor({ input, rate }: DropoutParams);
    propagate(input: Matrix): Matrix;
    backPropagate(_1: Matrix, _2: Matrix, loss: Matrix): Matrix;
}