@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
35 lines (34 loc) • 837 B
TypeScript
/**
* Naive bayes regression
*/
export default class NaiveBayesRegression {
/**
* @param {boolean[]} categoryPositions Category column position
*/
constructor(categoryPositions: boolean[]);
_iscat: boolean[];
_categories: any[];
_hx: any[];
_hy: any[];
_hk: any[];
_c_cand: number[];
_d: number;
_h: number;
_gaussian(x: any): number;
/**
* Fit model.
* @param {Array<Array<*>>} x Training data
* @param {Array<number>} y Target values
*/
fit(x: Array<Array<any>>, y: Array<number>): void;
_x: any[][];
_y: number[];
_ymax: any;
_ymin: any;
/**
* Returns predicted values.
* @param {Array<Array<*>>} x Sample data
* @returns {Array<number>} Predicted values
*/
predict(x: Array<Array<any>>): Array<number>;
}