@ai-on-browser/data-analysis-models/lib/model/forgetron.d.ts

Data analysis model package without any dependencies

/**
 * Forgetron
 */
export default class Forgetron {
    /**
     * @param {number} b Budget parameter
     * @param {'gaussian' | 'polynomial' | { name: 'gaussian', s?: number } | { name: 'polynomial', d?: number } | function (number[], number[]): number} [kernel] Kernel name
     */
    constructor(b: number, kernel?: "gaussian" | "polynomial" | {
        name: "gaussian";
        s?: number;
    } | {
        name: "polynomial";
        d?: number;
    } | ((arg0: number[], arg1: number[]) => number));
    _b: number;
    _kernel: any;
    _i: any[];
    _q: number;
    _m: number;
    _sigma: any[];
    /**
     * Fit model parameters.
     * @param {Array<Array<number>>} x Training data
     * @param {Array<1 | -1>} y Target values
     */
    fit(x: Array<Array<number>>, y: Array<1 | -1>): void;
    /**
     * Returns predicted values.
     * @param {Array<Array<number>>} data Sample data
     * @returns {(1 | -1)[]} Predicted values
     */
    predict(data: Array<Array<number>>): (1 | -1)[];
}