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

Data analysis model package without any dependencies

/**
 * Uniform Manifold Approximation and Projection
 */
export default class UMAP {
    /**
     * @param {number} rd Reduced dimension
     * @param {number} [n] Number of neighborhoods
     * @param {number} [min_dist] Minimum distance
     */
    constructor(rd: number, n?: number, min_dist?: number);
    _rd: number;
    _n: number;
    _min_dist: number;
    _neg_samples: number;
    /**
     * Initialize model.
     * @param {Array<Array<number>>} x Training data
     */
    init(x: Array<Array<number>>): void;
    _x: number[][][];
    _d(a: any, b: any): number;
    _init(): void;
    _w: Matrix<any>;
    _y: number[][];
    _alpha: number;
    _epoch: number;
    _a: number;
    _b: number;
    /**
     * Fit model and returns reduced values.
     * @returns {Array<Array<number>>} Predicted values
     */
    fit(): Array<Array<number>>;
    /**
     * Returns reduced values.
     * @returns {Array<Array<number>>} Predicted values
     */
    predict(): Array<Array<number>>;
}
import Matrix from '../util/matrix.js';