@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
36 lines (35 loc) • 797 B
TypeScript
/**
* Hartigan-Wong k-Means algorithm
*/
export default class HartiganWongKMeans {
/**
* @param {number} k Number of clusters
*/
constructor(k: number);
_k: number;
_c: any[];
_n: any[];
_d: (a: any, b: any) => number;
/**
* Centroids
* @type {Array<Array<number>>}
*/
get centroids(): number[][];
/**
* Initialize this model.
* @param {Array<Array<number>>} datas Training data
*/
init(datas: Array<Array<number>>): void;
_x: number[][];
_p: any[];
/**
* Fit model.
*/
fit(): void;
/**
* Returns predicted categories.
* @param {Array<Array<number>>} datas Sample data
* @returns {number[]} Predicted values
*/
predict(datas: Array<Array<number>>): number[];
}