class

CrystalML::Regression::BayesianRegression

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

Constructors

new(alpha : Float64 = 1.0, beta : Float64 = 1.0)

Instance methods

alpha

Hyperparameters for the prior distribution

alpha=(alpha : Float64)

Hyperparameters for the prior distribution

beta

beta=(beta : Float64)

covariance

covariance=(covariance : Tensor(Float64, CPU(Float64)))

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

mean

mean=(mean : Tensor(Float64, CPU(Float64)))

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

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