UNPKG

@ai-on-browser/data-analysis-models

Version:

Data analysis model package without any dependencies

66 lines (65 loc) 1.67 kB
/** * Stochastic Neighbor Embedding */ export class SNE { /** * @param {number} [rd] Reduced dimension * @param {number} [perplexity] Perplexity */ constructor(rd?: number, perplexity?: number); _epoch: number; _rd: number; _learning_rate: number; _perplexity: number; /** * Initialize model. * @param {Array<Array<number>>} x Training data */ init(x: Array<Array<number>>): void; _x: number[][]; _y: any[]; _pj_i(x: any): any[][]; /** * 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>>; } /** * T-distributed Stochastic Neighbor Embedding */ export class tSNE { /** * @param {Array<Array<number>>} datas Training data * @param {number} [rd] Reduced dimension * @param {number} [perplexity] Perplexity */ constructor(rd?: number, perplexity?: number); _epoch: number; _rd: number; _learning_rate: number; _perplexity: number; /** * Initialize model. * @param {Array<Array<number>>} x Training data */ init(x: Array<Array<number>>): void; _x: number[][]; _y: any[]; _pj_i(x: any): any[][]; /** * 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>>; }