@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
36 lines (35 loc) • 894 B
TypeScript
/**
* Universal-set Naive bayes
*/
export default class UniversalSetNaiveBayes {
/**
* @param {'gaussian'} [distribution] Distribution name
*/
constructor(distribution?: 'gaussian');
_labels: any[];
_rate: any[];
_ratec: any[];
_p_class: typeof Gaussian;
_p: any[];
_pc: any[];
/**
* Fit model.
* @param {Array<Array<number>>} datas Training data
* @param {*[]} labels Target values
*/
fit(datas: Array<Array<number>>, labels: any[]): void;
/**
* Returns predicted categories.
* @param {Array<Array<number>>} data Sample data
* @returns {*[]} Predicted values
*/
predict(data: Array<Array<number>>): any[];
}
declare class Gaussian {
_means: any;
_vars: any;
_estimate_prob(x: any): void;
_data_prob(x: any): Matrix<number>;
}
import Matrix from '../util/matrix.js';
export {};