UNPKG

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

Version:

Data analysis model package without any dependencies

30 lines (29 loc) 691 B
/** * MacQueen's k-Means algorithm */ export default class MacQueenKMeans { /** * @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[][]; /** * Fit model. * @param {Array<Array<number>>} datas Training data */ fit(datas: Array<Array<number>>): void; /** * Returns predicted categories. * @param {Array<Array<number>>} datas Sample data * @returns {number[]} Predicted values */ predict(datas: Array<Array<number>>): number[]; }