@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
31 lines (30 loc) • 1.06 kB
TypeScript
/**
* Nearest centroid classifier
*/
export default class NearestCentroid {
/**
* @param {'euclid' | 'manhattan' | 'chebyshev' | 'minkowski' | function (number[], number[]): number} [metric] Metric name
*/
constructor(metric?: "euclid" | "manhattan" | "chebyshev" | "minkowski" | ((arg0: number[], arg1: number[]) => number));
_c: any[];
_metric: "euclid" | "manhattan" | "chebyshev" | "minkowski" | ((arg0: number[], arg1: number[]) => number);
_d: (a: any, b: any) => any;
/**
* Add a data.
* @param {number[]} point Training data
* @param {*} category Target value
*/
add(point: number[], category: any): void;
/**
* Add datas.
* @param {Array<Array<number>>} datas Training data
* @param {*[]} targets Target values
*/
fit(datas: Array<Array<number>>, targets: any[]): void;
/**
* Returns predicted categories.
* @param {Array<Array<number>>} datas Sample data
* @returns {*[]} Predicted values
*/
predict(datas: Array<Array<number>>): any[];
}