Struct PoissonRegressor 

pub struct PoissonRegressor<F> {
    pub alpha: F,
    pub max_iter: usize,
    pub tol: F,
    pub fit_intercept: bool,
}

Poisson regressor — GLM with Poisson family and log link.

Suitable for modelling count data (y >= 0, integer-valued).

Type Parameters

• F: The floating-point type (f32 or f64).

Fields

alpha: F

L2 regularization strength.

max_iter: usize

Maximum number of IRLS iterations.

tol: F

Convergence tolerance.

fit_intercept: bool

Whether to fit an intercept.

Implementations

impl<F: Float + FromPrimitive> PoissonRegressor<F>

pub fn new() -> Self

Create a new PoissonRegressor with default settings.

Defaults: alpha = 1.0, max_iter = 100, tol = 1e-4, fit_intercept = true.

pub fn with_alpha(self, alpha: F) -> Self

Set the L2 regularization strength.

pub fn with_max_iter(self, max_iter: usize) -> Self

Set the maximum number of IRLS iterations.

pub fn with_tol(self, tol: F) -> Self

Set the convergence tolerance.

pub fn with_fit_intercept(self, fit_intercept: bool) -> Self

Set whether to fit an intercept.

Trait Implementations

impl<F: Clone> Clone for PoissonRegressor<F>

fn clone(&self) -> PoissonRegressor<F>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<F: Debug> Debug for PoissonRegressor<F>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<F: Float + FromPrimitive> Default for PoissonRegressor<F>

fn default() -> Self

Returns the “default value” for a type.

impl<F: Float + Send + Sync + ScalarOperand + FromPrimitive + 'static> Fit<ArrayBase<OwnedRepr<F>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<F>, Dim<[usize; 1]>>> for PoissonRegressor<F>

fn fit( &self, x: &Array2<F>, y: &Array1<F>, ) -> Result<FittedGLMRegressor<F>, FerroError>

Fit the Poisson GLM via IRLS.

Errors

See GLMRegressor::fit.

type Fitted = FittedGLMRegressor<F>

The fitted model type returned by fit.

type Error = FerroError

The error type returned by fit.

impl<F> PipelineEstimator<F> for PoissonRegressor<F>

where F: Float + FromPrimitive + ScalarOperand + Send + Sync + 'static,

fn fit_pipeline( &self, x: &Array2<F>, y: &Array1<F>, ) -> Result<Box<dyn FittedPipelineEstimator<F>>, FerroError>

Fit this estimator on the given data.