pub struct BayesianLinearRegression {
pub prior_mean: Array1<f64>,
pub prior_precision: Array2<f64>,
pub prior_alpha: f64,
pub prior_beta: f64,
pub fit_intercept: bool,
}Expand description
Bayesian linear regression with normal-inverse-gamma prior
This implements Bayesian linear regression where:
- Coefficients have a normal prior
- Noise variance has an inverse-gamma prior
- Posterior is analytically tractable
Fields§
§prior_mean: Array1<f64>Prior mean for coefficients
prior_precision: Array2<f64>Prior precision matrix for coefficients
prior_alpha: f64Prior shape parameter for noise variance
prior_beta: f64Prior scale parameter for noise variance
fit_intercept: boolWhether to include intercept
Implementations§
Source§impl BayesianLinearRegression
impl BayesianLinearRegression
Sourcepub fn new(n_features: usize, fit_intercept: bool) -> StatsResult<Self>
pub fn new(n_features: usize, fit_intercept: bool) -> StatsResult<Self>
Create a new Bayesian linear regression model
Sourcepub fn with_priors(
prior_mean: Array1<f64>,
prior_precision: Array2<f64>,
prior_alpha: f64,
prior_beta: f64,
fit_intercept: bool,
) -> StatsResult<Self>
pub fn with_priors( prior_mean: Array1<f64>, prior_precision: Array2<f64>, prior_alpha: f64, prior_beta: f64, fit_intercept: bool, ) -> StatsResult<Self>
Create with custom priors
Sourcepub fn fit(
&self,
x: ArrayView2<'_, f64>,
y: ArrayView1<'_, f64>,
) -> StatsResult<BayesianRegressionResult>
pub fn fit( &self, x: ArrayView2<'_, f64>, y: ArrayView1<'_, f64>, ) -> StatsResult<BayesianRegressionResult>
Fit the Bayesian regression model
Sourcepub fn predict(
&self,
x: ArrayView2<'_, f64>,
result: &BayesianRegressionResult,
) -> StatsResult<BayesianPredictionResult>
pub fn predict( &self, x: ArrayView2<'_, f64>, result: &BayesianRegressionResult, ) -> StatsResult<BayesianPredictionResult>
Make predictions on new data
Sourcepub fn predict_with_credible_interval(
&self,
x: ArrayView2<'_, f64>,
result: &BayesianRegressionResult,
confidence: f64,
) -> StatsResult<BayesianPredictionResult>
pub fn predict_with_credible_interval( &self, x: ArrayView2<'_, f64>, result: &BayesianRegressionResult, confidence: f64, ) -> StatsResult<BayesianPredictionResult>
Compute credible intervals for predictions
Trait Implementations§
Source§impl Clone for BayesianLinearRegression
impl Clone for BayesianLinearRegression
Source§fn clone(&self) -> BayesianLinearRegression
fn clone(&self) -> BayesianLinearRegression
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for BayesianLinearRegression
impl RefUnwindSafe for BayesianLinearRegression
impl Send for BayesianLinearRegression
impl Sync for BayesianLinearRegression
impl Unpin for BayesianLinearRegression
impl UnsafeUnpin for BayesianLinearRegression
impl UnwindSafe for BayesianLinearRegression
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
impl<T> Read<Exclusive, BecauseExclusive> for Twhere
T: ?Sized,
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.