pub struct MarginalLikelihoodOptimizer { /* private fields */ }Expand description
let X = array![[1.0], [2.0], [3.0], [4.0]]; let y = array![1.0, 4.0, 9.0, 16.0];
let mut kernel: Box
let result = optimizer.optimize( &X.view(), &y.view(), &mut kernel, 0.1 ).unwrap();
let optimized_params = result.optimal_params; let final_lml = result.optimal_log_marginal_likelihood;
Implementations§
Source§impl MarginalLikelihoodOptimizer
impl MarginalLikelihoodOptimizer
Sourcepub fn learning_rate(self, learning_rate: f64) -> Self
pub fn learning_rate(self, learning_rate: f64) -> Self
Set the learning rate
Sourcepub fn adam_params(self, beta1: f64, beta2: f64, epsilon: f64) -> Self
pub fn adam_params(self, beta1: f64, beta2: f64, epsilon: f64) -> Self
Set Adam optimizer parameters
Sourcepub fn line_search(self, line_search: bool) -> Self
pub fn line_search(self, line_search: bool) -> Self
Enable or disable line search
Sourcepub fn optimize(
&self,
X: &ArrayView2<'_, f64>,
y: &ArrayView1<'_, f64>,
kernel: &mut Box<dyn Kernel>,
sigma_n: f64,
) -> SklResult<OptimizationResult>
pub fn optimize( &self, X: &ArrayView2<'_, f64>, y: &ArrayView1<'_, f64>, kernel: &mut Box<dyn Kernel>, sigma_n: f64, ) -> SklResult<OptimizationResult>
Optimize the log marginal likelihood
Trait Implementations§
Source§impl Clone for MarginalLikelihoodOptimizer
impl Clone for MarginalLikelihoodOptimizer
Source§fn clone(&self) -> MarginalLikelihoodOptimizer
fn clone(&self) -> MarginalLikelihoodOptimizer
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for MarginalLikelihoodOptimizer
impl Debug for MarginalLikelihoodOptimizer
Auto Trait Implementations§
impl Freeze for MarginalLikelihoodOptimizer
impl RefUnwindSafe for MarginalLikelihoodOptimizer
impl Send for MarginalLikelihoodOptimizer
impl Sync for MarginalLikelihoodOptimizer
impl Unpin for MarginalLikelihoodOptimizer
impl UnwindSafe for MarginalLikelihoodOptimizer
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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