@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
66 lines (65 loc) • 1.67 kB
TypeScript
/**
* 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>>;
}