class

CrystalML::Classification::NaiveBayesClassifier

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

Constructors

Instance methods

fit(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64)))
Source
means
Source
means=(means : Hash(Float64, Tensor(Float64, CPU(Float64))))
Source
predict(data : Tensor(Float64, CPU(Float64))) : Tensor(Float64, CPU(Float64))
Source
variances
Source
variances=(variances : Hash(Float64, Tensor(Float64, CPU(Float64))))
Source