@hoff97/tensor-js/dist/lib/model/regularization.d.ts

PyTorch like deep learning inferrence library

import { Variable } from '../autograd';
import { Module } from './module';
/**
 * L2 weight regularization for a particular model.
 *
 * @example
 * ```typescript
 * const model = new Linear(32,1);
 * const regularizer = new L2Regularization(model, 0.01);
 * //...
 * const prediction = (await model.forward([x]))[0];
 * let loss = prediction.subtract(y).reduceSumSquare();
 * loss = loss.add(regularizer.getLoss());
 * //...
 * loss.backward();
 * //...
 * ```
 */
export declare class L2Regularization {
    model: Module;
    gamma: number;
    parameters: Variable<any>[];
    constructor(model: Module, gamma: number);
    getLoss(): import("../types").default<any>;
}
/**
 * L1 weight regularization for a particular model.
 *
 * @example
 * ```typescript
 * const model = new Linear(32,1);
 * const regularizer = new L1Regularization(model, 0.01);
 * //...
 * const prediction = (await model.forward([x]))[0];
 * let loss = prediction.subtract(y).reduceSumSquare();
 * loss = loss.add(regularizer.getLoss());
 * //...
 * loss.backward();
 * //...
 * ```
 */
export declare class L1Regularization {
    model: Module;
    gamma: number;
    parameters: Variable<any>[];
    constructor(model: Module, gamma: number);
    getLoss(): import("../types").default<any>;
}