@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
21 lines (20 loc) • 483 B
TypeScript
/**
* Zero-inflated negative binomial
*/
export default class ZeroInflatedNegativeBinomial {
/**
* Fit model.
* @param {number[]} x Training data
*/
fit(x: number[]): void;
_pi: number;
_l: number;
_k: number;
_probability(z: any, p: any, k: any, l: any): any;
/**
* Returns predicted probabilities.
* @param {number[]} x Sample data
* @returns {number[]} Predicted values
*/
probability(x: number[]): number[];
}