Class CategoricalNaiveBayes

constructor

• new CategoricalNaiveBayes(alpha?: number): CategoricalNaiveBayes

• Parameters

  • Optionalalpha: number

    Smoothing parameter

  Returns CategoricalNaiveBayes

_alpha

_cand

_d

_labels

_prob

fit

• fit(datas: any[][], labels: any[]): void

• Fit model.

  Parameters

  • datas: any[][]

    Training data
  • labels: any[]

    Target values

  Returns void

predict

• predict(datas: any[][]): any[]

• Returns predicted categories.

  Parameters

  • datas: any[][]

    Sample data

  Returns any[]

  Predicted values

probability

• probability(datas: any[][]): number[][]

• Returns predicted probabilities.

  Parameters

  • datas: any[][]

    Sample data

  Returns number[][]

  Predicted values