class

CrystalML::Classification::NaiveBayesClassifier

Inherits CrystalML::Classifier / CrystalML::SupervisedEstimator / CrystalML::BaseEstimator / Reference / Object

Constructors

new

Instance methods

fit(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64)))

means

means=(means : Hash(Float64, Tensor(Float64, CPU(Float64))))

predict(data : Tensor(Float64, CPU(Float64))) : Tensor(Float64, CPU(Float64))

variances

variances=(variances : Hash(Float64, Tensor(Float64, CPU(Float64))))