UNPKG

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

Version:

Data analysis model package without any dependencies

36 lines (35 loc) 797 B
/** * 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[]; }