Crate variational_regression

This crate provides functionality for regression models trained using variational inference

Re-exports

pub use distribution::ScalarDistribution;

pub use distribution::GammaDistribution;

pub use distribution::GaussianDistribution;

pub use distribution::BernoulliDistribution;

pub use linear::VariationalLinearRegression;

pub use linear::LinearTrainConfig;

pub use logistic::VariationalLogisticRegression;

pub use logistic::LogisticTrainConfig;

pub use error::RegressionError;

Modules

distribution

error

linear

logistic

Traits

BinaryLabels

Represents binary label data

Features

Represents two dimensional, dense feature data

RealLabels

Represents continuous, real-valued label data

VariationalRegression

Represents a trained variational regression model with the specified predictive distribution type