pub struct Bart { /* private fields */ }Expand description
A fitted BART model: the retained posterior draws (each a forest of trees).
Implementations§
Source§impl Bart
impl Bart
Sourcepub fn fit(
x: &[f64],
y: &[f64],
n: usize,
p: usize,
m: usize,
n_iter: usize,
burn_in: usize,
k: f64,
seed: u64,
) -> Result<Self, LearningError>
pub fn fit( x: &[f64], y: &[f64], n: usize, p: usize, m: usize, n_iter: usize, burn_in: usize, k: f64, seed: u64, ) -> Result<Self, LearningError>
Fit BART. m trees, n_iter MCMC sweeps with burn_in discarded. k
controls the leaf prior (≈ 2). Fails closed on shape mismatch / degenerate y.
Sourcepub fn predict_row(&self, x_row: &[f64]) -> f64
pub fn predict_row(&self, x_row: &[f64]) -> f64
Posterior-mean prediction for one row (in the original y units).
pub fn predict(&self, x: &[f64], n: usize) -> Vec<f64>
pub fn n_draws(&self) -> usize
Trait Implementations§
Auto Trait Implementations§
impl Freeze for Bart
impl RefUnwindSafe for Bart
impl Send for Bart
impl Sync for Bart
impl Unpin for Bart
impl UnsafeUnpin for Bart
impl UnwindSafe for Bart
Blanket Implementations§
§impl<S, A> Aggregate<Result<S, Error>> for Awhere
A: Aggregate<S>,
impl<S, A> Aggregate<Result<S, Error>> for Awhere
A: Aggregate<S>,
Aggregate shares in an MPC protocol.
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§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 more