@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
41 lines (40 loc) • 910 B
TypeScript
/**
* Drake's accelerated k-Means algorithm
*/
export default class DrakeKMeans {
/**
* @param {number} k Number of clusters
*/
constructor(k: number);
_k: number;
_b: number;
_c: number[][];
_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[][];
_t: any[];
_y: any[];
_u: any[];
_a: any[];
_l: any[];
_sortCenters(i: any, q: any, r: any): void;
/**
* Fit model.
*/
fit(): void;
/**
* Returns predicted categories.
* @param {Array<Array<number>>} datas Sample data
* @returns {number[]} Predicted values
*/
predict(datas: Array<Array<number>>): number[];
}