Class NaiveBayes

constructor

• new NaiveBayes(distribution?: "gaussian"): NaiveBayes

• Parameters

  • Optionaldistribution: "gaussian"

    Distribution name

  Returns NaiveBayes

_labels

_p

_p_class

_rate

fit

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

• Fit model.

  Parameters

  • datas: number[][]

    Training data
  • labels: any[]

    Target values

  Returns void

predict

• predict(data: number[][]): any[]

• Returns predicted categories.

  Parameters

  • data: number[][]

    Sample data

  Returns any[]

  Predicted values

probability

• probability(data: number[][]): number[][]

• Returns predicted probabilities.

  Parameters

  • data: number[][]

    Sample data

  Returns number[][]

  Predicted values