bayes

package

v0.1.0 Latest Latest Go to latest Published: Oct 6, 2021 License: MIT Imports: 2 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Class = ml.Class

type Column = ml.Column

type DType = ml.DType

type NaiveBayes struct {
	// contains filtered or unexported fields
}

func New(classes []string, prob []DType, featuresDim int) *NaiveBayes

func NewFromDataset(data ml.Dataset) *NaiveBayes

func (nb *NaiveBayes) Fit(dataset [][]Column)

func (nb *NaiveBayes) PredictTo(probs []DType, object []DType)

func (nb *NaiveBayes) SetProb(prob []DType)

type Row = ml.Row