pub struct Lasso<F> {
pub alpha: F,
pub max_iter: usize,
pub tol: F,
pub fit_intercept: bool,
}Expand description
Lasso regression (L1-regularized least squares).
Uses coordinate descent with soft-thresholding to solve the L1-penalized
regression problem. The alpha parameter controls the strength of the
L1 penalty.
§Type Parameters
F: The floating-point type (f32orf64).
Fields§
§alpha: FRegularization strength. Larger values specify stronger regularization and sparser solutions.
max_iter: usizeMaximum number of coordinate descent iterations.
tol: FConvergence tolerance on the maximum coefficient change.
fit_intercept: boolWhether to fit an intercept (bias) term.
Implementations§
Source§impl<F: Float> Lasso<F>
impl<F: Float> Lasso<F>
Sourcepub fn new() -> Self
pub fn new() -> Self
Create a new Lasso with default settings.
Defaults: alpha = 1.0, max_iter = 1000, tol = 1e-4,
fit_intercept = true.
Sourcepub fn with_alpha(self, alpha: F) -> Self
pub fn with_alpha(self, alpha: F) -> Self
Set the regularization strength.
Sourcepub fn with_max_iter(self, max_iter: usize) -> Self
pub fn with_max_iter(self, max_iter: usize) -> Self
Set the maximum number of iterations.
Sourcepub fn with_fit_intercept(self, fit_intercept: bool) -> Self
pub fn with_fit_intercept(self, fit_intercept: bool) -> Self
Set whether to fit an intercept term.
Trait Implementations§
Source§impl<F: Float + Send + Sync + ScalarOperand + FromPrimitive + 'static> Fit<ArrayBase<OwnedRepr<F>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<F>, Dim<[usize; 1]>>> for Lasso<F>
impl<F: Float + Send + Sync + ScalarOperand + FromPrimitive + 'static> Fit<ArrayBase<OwnedRepr<F>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<F>, Dim<[usize; 1]>>> for Lasso<F>
Source§fn fit(
&self,
x: &Array2<F>,
y: &Array1<F>,
) -> Result<FittedLasso<F>, FerroError>
fn fit( &self, x: &Array2<F>, y: &Array1<F>, ) -> Result<FittedLasso<F>, FerroError>
Fit the Lasso model using coordinate descent.
§Errors
Returns FerroError::ShapeMismatch if the number of samples in
x and y differ.
Returns FerroError::InvalidParameter if alpha is negative.
Returns FerroError::ConvergenceFailure if the algorithm does
not converge within max_iter iterations.
Source§type Fitted = FittedLasso<F>
type Fitted = FittedLasso<F>
fit.Source§type Error = FerroError
type Error = FerroError
fit.Source§impl<F> PipelineEstimator<F> for Lasso<F>
impl<F> PipelineEstimator<F> for Lasso<F>
Source§fn fit_pipeline(
&self,
x: &Array2<F>,
y: &Array1<F>,
) -> Result<Box<dyn FittedPipelineEstimator<F>>, FerroError>
fn fit_pipeline( &self, x: &Array2<F>, y: &Array1<F>, ) -> Result<Box<dyn FittedPipelineEstimator<F>>, FerroError>
Auto Trait Implementations§
impl<F> Freeze for Lasso<F>where
F: Freeze,
impl<F> RefUnwindSafe for Lasso<F>where
F: RefUnwindSafe,
impl<F> Send for Lasso<F>where
F: Send,
impl<F> Sync for Lasso<F>where
F: Sync,
impl<F> Unpin for Lasso<F>where
F: Unpin,
impl<F> UnsafeUnpin for Lasso<F>where
F: UnsafeUnpin,
impl<F> UnwindSafe for Lasso<F>where
F: UnwindSafe,
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> DistributionExt for Twhere
T: ?Sized,
impl<T> DistributionExt for Twhere
T: ?Sized,
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>
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>
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