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

Data analysis model package without any dependencies

/**
 * Gradient boosting decision tree
 */
export class GBDT {
    /**
     * @param {number} [maxdepth] Maximum depth of tree
     * @param {number} [srate] Sampling rate
     * @param {number} [lr] Learning rate
     */
    constructor(maxdepth?: number, srate?: number, lr?: number);
    _trees: any[];
    _r: any[];
    _maxd: number;
    _srate: number;
    _lr: number;
    /**
     * Number of trees
     * @type {number}
     */
    get size(): number;
    _sample(n: any): number[];
    /**
     * Initialize model.
     * @param {Array<Array<number>>} x Training data
     * @param {Array<Array<number>>} y Target values
     */
    init(x: Array<Array<number>>, y: Array<Array<number>>): void;
    _x: number[][];
    _y: Matrix<number[]>;
    _loss: Matrix<number[]>;
    /**
     * Fit model.
     */
    fit(): void;
    /**
     * Returns predicted values.
     * @param {Array<Array<number>>} x Sample data
     * @returns {Array<Array<number>>} Predicted values
     */
    predict(x: Array<Array<number>>): Array<Array<number>>;
}
/**
 * Gradient boosting decision tree classifier
 */
export class GBDTClassifier extends GBDT {
    /**
     * Initialize model.
     * @param {Array<Array<number>>} x Training data
     * @param {*[]} y Target values
     */
    init(x: Array<Array<number>>, y: any[]): void;
    _cls: any[];
    /**
     * Returns predicted categories.
     * @param {Array<Array<number>>} x Sample data
     * @returns {*[]} Predicted values
     */
    predict(x: Array<Array<number>>): any[];
}
import Matrix from '../util/matrix.js';