chemotools 0.1.9__py3-none-any.whl → 0.1.10__py3-none-any.whl

chemotools/augmentation/__init__.py

@@ -1,6 +1,7 @@
 from ._add_noise import AddNoise
 from ._baseline_shift import BaselineShift
 from ._fractional_shift import FractionalShift
+from ._gaussian_broadening import GaussianBroadening
 from ._index_shift import IndexShift
 from ._spectrum_scale import SpectrumScale
 
@@ -9,6 +10,7 @@ __all__ = [
     "AddNoise",
     "BaselineShift",
     "FractionalShift",
+    "GaussianBroadening",
     "IndexShift",
     "SpectrumScale",
 ]

chemotools/augmentation/_gaussian_broadening.py

@@ -0,0 +1,136 @@
+from typing import Literal, Optional
+import numpy as np
+from scipy.ndimage import gaussian_filter1d
+from sklearn.base import BaseEstimator, TransformerMixin, OneToOneFeatureMixin
+from sklearn.utils.validation import check_is_fitted, validate_data
+
+
+class GaussianBroadening(TransformerMixin, OneToOneFeatureMixin, BaseEstimator):
+    """
+    Transform spectral data by broadening peaks using Gaussian convolution.
+
+    This transformer applies Gaussian smoothing to broaden peaks in spectral data.
+    For each signal, a random sigma is chosen between 0 and the specified sigma value.
+
+    Parameters
+    ----------
+    sigma : float, default=1.0
+        Maximum standard deviation for the Gaussian kernel.
+        The actual sigma used will be randomly chosen between 0 and this value.
+
+    mode : {'reflect', 'constant', 'nearest', 'mirror', 'wrap'}, default='reflect'
+        The mode parameter determines how the input array is extended when
+        the filter overlaps a border. Default is 'reflect'.
+
+    pad_value : float, default=0.0
+        Value to fill past edges of input if mode is 'constant'.
+
+    random_state : int, optional, default=None
+        Random state for reproducible sigma selection.
+
+    truncate : float, default=4.0
+        Truncate the filter at this many standard deviations.
+        Larger values increase computation time but improve accuracy.
+    """
+
+    def __init__(
+        self,
+        sigma: float = 1.0,
+        mode: Literal["reflect", "constant", "nearest", "mirror", "wrap"] = "reflect",
+        pad_value: float = 0.0,
+        random_state: Optional[int] = None,
+        truncate: float = 4.0,
+    ):
+        self.sigma = sigma
+        self.mode = mode
+        self.pad_value = pad_value
+        self.random_state = random_state
+        self.truncate = truncate
+
+    def fit(self, X: np.ndarray, y=None) -> "GaussianBroadening":
+        """
+        Fit the transformer to the data (in this case, only validates input).
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data to validate.
+
+        y : None
+            Ignored.
+
+        Returns
+        -------
+        self : GaussianBroadening
+            The fitted transformer.
+        """
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        # Validate sigma parameter
+        if not isinstance(self.sigma, (int, float)):
+            raise ValueError("sigma must be a number")
+        if self.sigma < 0:
+            raise ValueError("sigma must be non-negative")
+
+        # Initialize random number generator
+        self._rng = np.random.default_rng(self.random_state)
+
+        return self
+
+    def transform(self, X: np.ndarray, y=None) -> np.ndarray:
+        """
+        Apply Gaussian broadening to the input data.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            The data to transform.
+
+        y : None
+            Ignored.
+
+        Returns
+        -------
+        X_transformed : ndarray of shape (n_samples, n_features)
+            The transformed data with broadened peaks.
+        """
+        check_is_fitted(self, "n_features_in_")
+        X_ = validate_data(
+            self,
+            X,
+            y="no_validation",
+            ensure_2d=True,
+            copy=True,
+            reset=False,
+            dtype=np.float64,
+        )
+
+        # Transform each sample
+        for i, x in enumerate(X_):
+            X_[i] = self._broaden_signal(x)
+
+        return X_
+
+    def _broaden_signal(self, x: np.ndarray) -> np.ndarray:
+        """
+        Apply Gaussian broadening to a single signal.
+
+        Parameters
+        ----------
+        x : ndarray of shape (n_features,)
+            The input signal to broaden.
+
+        Returns
+        -------
+        broadened_signal : ndarray of shape (n_features,)
+            The broadened signal.
+        """
+        # Randomly choose sigma between 0 and max sigma
+        sigma = self._rng.uniform(0, self.sigma)
+
+        # Apply Gaussian filter
+        return gaussian_filter1d(
+            x, sigma=sigma, mode=self.mode, cval=self.pad_value, truncate=self.truncate
+        )

chemotools/outliers/__init__.py

@@ -0,0 +1,7 @@
+from .dmodx import DModX
+from .hotelling_t2 import HotellingT2
+from .q_residuals import QResiduals
+from .leverage import Leverage
+from .studentized_residuals import StudentizedResiduals
+
+__all__ = ["DModX", "HotellingT2", "QResiduals", "Leverage", "StudentizedResiduals"]

chemotools/outliers/_base.py

@@ -0,0 +1,180 @@
+from abc import ABC, abstractmethod
+from typing import Union, Optional
+
+import numpy as np
+
+from sklearn.base import BaseEstimator, OutlierMixin
+from sklearn.decomposition._base import _BasePCA
+from sklearn.cross_decomposition._pls import _PLS
+from sklearn.pipeline import Pipeline
+from sklearn.utils.validation import check_is_fitted
+
+from ._utils import validate_confidence, validate_and_extract_model
+
+ModelTypes = Union[_BasePCA, _PLS]
+
+
+class _ModelResidualsBase(ABC, BaseEstimator, OutlierMixin):
+    """Base class for model outlier calculations.
+
+    Implements statistical calculations for outlier detection in dimensionality
+    reduction models like PCA and PLS.
+
+    Parameters
+    ----------
+    model : Union[ModelTypes, Pipeline]
+        A fitted _BasePCA or _PLS models or Pipeline ending with such a model
+    confidence : float
+        Confidence level for statistical calculations (between 0 and 1)
+
+    Attributes
+    ----------
+    model_ : ModelTypes
+        The fitted model of type _BasePCA or _PLS
+
+    preprocessing_ : Optional[Pipeline]
+        Preprocessing steps before the model
+
+    n_features_in_ : int
+        Number of features in the input data
+
+    n_components_ : int
+        Number of components in the model
+
+    n_samples_ : int
+        Number of samples used to train the model
+
+    critical_value_ : float
+        The calculated critical value for outlier detection
+    """
+
+    def __init__(
+        self,
+        model: Union[ModelTypes, Pipeline],
+        confidence: float,
+    ) -> None:
+        (
+            self.model_,
+            self.preprocessing_,
+            self.n_features_in_,
+            self.n_components_,
+            self.n_samples_,
+        ) = validate_and_extract_model(model)
+        self.confidence = validate_confidence(confidence)
+
+    def fit_predict_residuals(
+        self, X: np.ndarray, y: Optional[np.ndarray] = None
+    ) -> np.ndarray:
+        """Fit the model to the input data and calculate the residuals.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        y : array-like of shape (n_samples,), default=None
+            Target values
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            The residuals of the model
+        """
+        self.fit(X, y)
+        return self.predict_residuals(X, y, validate=True)
+
+    @abstractmethod
+    def predict_residuals(
+        self, X: np.ndarray, y: Optional[np.ndarray], validate: bool
+    ) -> np.ndarray:
+        """Calculate the residuals of the model.
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            The residuals of the model
+        """
+
+    @abstractmethod
+    def _calculate_critical_value(self, X: Optional[np.ndarray]) -> float:
+        """Calculate the critical value for outlier detection.
+
+        Returns
+        -------
+        float
+            The calculated critical value for outlier detection
+        """
+
+
+class _ModelDiagnosticsBase(ABC):
+    """Base class for model diagnostics methods. This does not implement outlier detection algorithms,
+    but rather implements methods that are used to assess trained models.
+
+    Parameters
+    ----------
+    model : Union[ModelTypes, Pipeline]
+        A fitted PCA/PLS model or Pipeline ending with such a model
+
+    Attributes
+    ----------
+    model_ : ModelTypes
+        The fitted model of type _BasePCA or _PLS
+
+    preprocessing_ : Optional[Pipeline]
+        Preprocessing steps before the model
+
+    """
+
+    def __init__(self, model: Union[ModelTypes, Pipeline]):
+        self.model_, self.preprocessing_ = self._validate_and_extract_model(model)
+
+    def _validate_and_extract_model(self, model):
+        """Validate and extract the model and preprocessing steps.
+
+        Parameters
+        ----------
+        model : Union[ModelTypes, Pipeline]
+            A fitted PCA/PLS model or Pipeline ending with such a model
+
+        Returns
+        -------
+        Tuple[ModelTypes, Optional[Pipeline]]
+            The extracted model and preprocessing steps
+
+        Raises
+        ------
+        ValueError
+            If the model is not of type _BasePCA or _PLS or a Pipeline ending with one of these types or if the model is not fitted
+        """
+        if isinstance(model, Pipeline):
+            preprocessing = model[:-1]
+            model = model[-1]
+        else:
+            preprocessing = None
+
+        if isinstance(model, (_BasePCA, _PLS)):
+            check_is_fitted(model)
+        else:
+            raise ValueError(
+                "Model not a valid model. Must be of base type _BasePCA or _PLS or a Pipeline ending with one of these types."
+            )
+        check_is_fitted(model)
+        return model, preprocessing
+
+    @abstractmethod
+    def predict(self, X: np.ndarray, y: Optional[np.ndarray]) -> np.ndarray:
+        """Predict the output of the model.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        y : array-like of shape (n_samples,), default=None
+            Target values
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Predicted values
+        """

chemotools/outliers/_utils.py

@@ -0,0 +1,91 @@
+from typing import Optional, Tuple, Union
+
+from sklearn.cross_decomposition._pls import _PLS
+from sklearn.decomposition._base import _BasePCA
+from sklearn.pipeline import Pipeline
+from sklearn.utils.validation import check_is_fitted
+
+ModelTypes = Union[_BasePCA, _PLS]
+
+
+def get_model_parameters(model: ModelTypes) -> Tuple[int, int, int]:
+    """
+    Get the number of features, components and samples from a model with PLS or PCA. types.
+
+    Parameters
+    ----------
+    model : ModelType
+        A fitted model of type _BasePCA or _PLS
+
+    Returns
+    -------
+    Tuple[int, int, int]
+        The number of features, components and samples in the model
+    """
+    if isinstance(model, _BasePCA):
+        return model.n_features_in_, model.n_components_, model.n_samples_
+    elif isinstance(model, _PLS):
+        return model.n_features_in_, model.n_components, len(model.x_scores_)
+    else:
+        raise ValueError(
+            "Model not a valid model. Must be of base type _BasePCA or _PLS or a Pipeline ending with one of these types."
+        )
+
+
+def validate_confidence(confidence: float) -> float:
+    """Validate parameters using sklearn conventions.
+
+    Parameters
+    ----------
+    confidence : float
+        Confidence level for statistical calculations (between 0 and 1)
+
+    Returns
+    -------
+    float
+        The validated confidence level
+
+    Raises
+    ------
+    ValueError
+        If confidence is not between 0 and 1
+    """
+    if not 0 < confidence < 1:
+        raise ValueError("Confidence must be between 0 and 1")
+    return confidence
+
+
+def validate_and_extract_model(
+    model: Union[ModelTypes, Pipeline],
+) -> Tuple[ModelTypes, Optional[Pipeline], int, int, int]:
+    """Validate and extract the model and preprocessing steps.
+
+    Parameters
+    ----------
+    model : Union[ModelTypes, Pipeline]
+        A fitted PCA/PLS model or Pipeline ending with such a model
+
+    Returns
+    -------
+    Tuple[ModelTypes, Optional[Pipeline]]
+        The extracted model and preprocessing steps
+
+    Raises
+    ------
+    ValueError
+        If the model is not of type _BasePCA or _PLS or a Pipeline ending with one of these types or if the model is not fitted
+    """
+    if isinstance(model, Pipeline):
+        preprocessing = model[:-1]
+        model = model[-1]
+    else:
+        preprocessing = None
+
+    if not isinstance(model, (_BasePCA, _PLS)):
+        raise ValueError(
+            "Model not a valid model. Must be of base type _BasePCA or _PLS or a Pipeline ending with one of these types."
+        )
+
+    check_is_fitted(model)
+    n_features_in, n_components, n_samples = get_model_parameters(model)
+    return model, preprocessing, n_features_in, n_components, n_samples

chemotools/outliers/dmodx.py

@@ -0,0 +1,146 @@
+from typing import Optional, Union
+import numpy as np
+
+from sklearn.pipeline import Pipeline
+from sklearn.utils.validation import validate_data, check_is_fitted
+from scipy.stats import f as f_distribution
+
+
+from ._base import _ModelResidualsBase, ModelTypes
+
+
+class DModX(_ModelResidualsBase):
+    """Calculate Distance to Model (DModX) statistics.
+
+    DModX measures the distance between an observation and the model plane
+    in the X-space, useful for detecting outliers.
+
+    Parameters
+    ----------
+    model : Union[ModelType, Pipeline]
+        A fitted PCA/PLS model or Pipeline ending with such a model
+
+    confidence : float, default=0.95
+        Confidence level for statistical calculations (between 0 and 1)
+
+    Attributes
+    ----------
+    model_ : ModelType
+        The fitted model of type _BasePCA or _PLS
+
+    preprocessing_ : Optional[Pipeline]
+        Preprocessing steps before the model
+
+    n_features_in_ : int
+        Number of features in the input data
+
+    n_components_ : int
+        Number of components in the model
+
+    n_samples_ : int
+        Number of samples used to train the model
+
+    critical_value_ : float
+        The calculated critical value for outlier detection
+    """
+
+    def __init__(
+        self,
+        model: Union[ModelTypes, Pipeline],
+        confidence: float = 0.95,
+    ) -> None:
+        super().__init__(model, confidence)
+
+    def fit(self, X: np.ndarray, y: Optional[np.ndarray] = None) -> "DModX":
+        """
+        Fit the model to the input data.
+
+        This step calculates the critical value for the outlier detection. In the DmodX method,
+        the critical value is not depend on the input data but on the model parameters.
+        """
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        self.critical_value_ = self._calculate_critical_value()
+        return self
+
+    def predict(self, X: np.ndarray) -> np.ndarray:
+        """Identify outliers in the input data.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Boolean array indicating outliers
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        # Calculate outliers based on the DModX statistics
+        dmodx_values = self.predict_residuals(X, validate=False)
+        return np.where(dmodx_values > self.critical_value_, -1, 1)
+
+    def predict_residuals(
+        self, X: np.ndarray, y: Optional[np.ndarray] = None, validate: bool = True
+    ) -> np.ndarray:
+        """Calculate DModX statistics for input data.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        validate : bool, default=True
+            Whether to validate the input data
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            DModX statistics for each sample
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        if validate:
+            X = validate_data(
+                self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+            )
+
+        # Apply preprocessing if available
+        if self.preprocessing_:
+            X = self.preprocessing_.transform(X)
+
+        # Calculate the DModX statistics
+        X_transformed = self.model_.transform(X)
+        X_reconstructed = self.model_.inverse_transform(X_transformed)
+        squared_errors = np.sum((X - X_reconstructed) ** 2, axis=1)
+
+        return np.sqrt(squared_errors / (self.n_features_in_ - self.n_components_))
+
+    def _calculate_critical_value(self, X: Optional[np.ndarray] = None) -> float:
+        """Calculate F-distribution based critical value.
+
+        Returns
+        -------
+        float
+            The critical value for outlier detection
+        """
+
+        dof_numerator = self.n_features_in_ - self.n_components_
+        dof_denominator = self.n_features_in_ - self.n_components_ - 1
+
+        upper_control_limit = f_distribution.ppf(
+            self.confidence, dof_numerator, dof_denominator
+        )
+        return np.sqrt(upper_control_limit)

chemotools/outliers/hotelling_t2.py

@@ -0,0 +1,155 @@
+from typing import Optional, Union
+import numpy as np
+
+from sklearn.cross_decomposition._pls import _PLS
+from sklearn.decomposition._base import _BasePCA
+from sklearn.pipeline import Pipeline
+from sklearn.utils.validation import validate_data, check_is_fitted
+from scipy.stats import f as f_distribution
+
+from ._base import _ModelResidualsBase, ModelTypes
+
+
+class HotellingT2(_ModelResidualsBase):
+    """
+    Calculate Hotelling's T-squared statistics for PCA or PLS like models.
+
+    Parameters
+    ----------
+    model : Union[ModelType, Pipeline]
+        A fitted PCA/PLS model or Pipeline ending with such a model
+
+    confidence : float, default=0.95
+        Confidence level for statistical calculations (between 0 and 1)
+
+    Attributes
+    ----------
+    model_ : ModelType
+        The fitted model of type _BasePCA or _PLS
+
+    preprocessing_ : Optional[Pipeline]
+        Preprocessing steps before the model
+
+    n_features_in_ : int
+        Number of features in the input data
+
+    n_components_ : int
+        Number of components in the model
+
+    n_samples_ : int
+        Number of samples used to train the model
+
+    critical_value_ : float
+        The calculated critical value for outlier detection
+
+    References
+    ----------
+    Johan A. Westerhuis, Stephen P. Gurden, Age K. Smilde (2001) Generalized contribution plots in multivariate statistical process
+    monitoring  Chemometrics and Intelligent Laboratory Systems 51 2000 95–114
+    """
+
+    def __init__(
+        self, model: Union[ModelTypes, Pipeline], confidence: float = 0.95
+    ) -> None:
+        super().__init__(model, confidence)
+
+    def fit(self, X: np.ndarray, y: Optional[np.ndarray] = None) -> "HotellingT2":
+        """
+        Fit the model to the input data.
+
+        This step calculates the critical value for the outlier detection. In the DmodX method,
+        the critical value is not depend on the input data but on the model parameters.
+        """
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        self.critical_value_ = self._calculate_critical_value()
+        return self
+
+    def predict(self, X: np.ndarray) -> np.ndarray:
+        """Identify outliers in the input data.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Boolean array indicating outliers
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        # Calculate the Hotelling's T-squared statistics
+        hotelling_t2_values = self.predict_residuals(X, y=None, validate=False)
+        return np.where(hotelling_t2_values > self.critical_value_, -1, 1)
+
+    def predict_residuals(
+        self, X: np.ndarray, y: Optional[np.ndarray], validate: bool = True
+    ) -> np.ndarray:
+        """Calculate Hotelling's T-squared statistics for input data.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Hotellin's T-squared statistics for each sample
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        if validate:
+            X = validate_data(
+                self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+            )
+
+        # Apply preprocessing steps
+        if self.preprocessing_:
+            X = self.preprocessing_.transform(X)
+
+        # Calculate the Hotelling's T-squared statistics
+        if isinstance(self.model_, _BasePCA):
+            # For PCA-like models
+            variances = self.model_.explained_variance_
+
+        if isinstance(self.model_, _PLS):
+            # For PLS-like models
+            variances = np.var(self.model_.x_scores_, axis=0)
+
+        # Equivalent to X @ model.components_.T for _BasePCA and X @ model.x_rotations_ for _PLS
+        X_transformed = self.model_.transform(X)
+
+        return np.sum((X_transformed**2) / variances, axis=1)
+
+    def _calculate_critical_value(self, X: Optional[np.ndarray] = None) -> float:
+        """
+        Calculate the critical value for the Hotelling's T-squared statistics.
+
+        Returns
+        -------
+        float
+            The critical value for the Hotelling's T-squared statistics
+        """
+
+        critical_value = f_distribution.ppf(
+            self.confidence, self.n_components_, self.n_samples_ - self.n_components_
+        )
+        return (
+            critical_value
+            * self.n_components_
+            * (self.n_samples_ - 1)
+            / (self.n_samples_ - self.n_components_)
+        )

chemotools/outliers/leverage.py

@@ -0,0 +1,150 @@
+from typing import Optional, Union
+import numpy as np
+
+from sklearn.pipeline import Pipeline
+from sklearn.utils.validation import validate_data, check_is_fitted
+
+
+from ._base import _ModelResidualsBase, ModelTypes
+
+
+class Leverage(_ModelResidualsBase):
+    """
+    Calculate the leverage of the training samples on  the latent space of a PCA or PLS models.
+    This method allows to detect datapoints with high leverage in the model.
+
+    Parameters
+    ----------
+    model : Union[ModelType, Pipeline]
+        A fitted PCA/PLS model or Pipeline ending with such a model
+
+    Attributes
+    ----------
+    model_ : ModelType
+        The fitted model of type _BasePCA or _PLS
+
+    preprocessing_ : Optional[Pipeline]
+        Preprocessing steps before the model
+
+    References
+    ----------
+
+    """
+
+    def __init__(
+        self, model: Union[ModelTypes, Pipeline], confidence: float = 0.95
+    ) -> None:
+        super().__init__(model, confidence)
+
+    def fit(self, X: np.ndarray, y: Optional[np.ndarray] = None) -> "Leverage":
+        """
+        Fit the model to the input data.
+
+        Parameters
+
+        """
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        if self.preprocessing_:
+            X = self.preprocessing_.fit_transform(X)
+
+        # Compute the critical threshold
+        self.critical_value_ = self._calculate_critical_value(X)
+
+        return self
+
+    def predict(self, X: np.ndarray, y: Optional[np.ndarray] = None) -> np.ndarray:
+        """Calculate Leverage for training data on the model.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Bool with samples with a leverage above the critical value
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        # Preprocess the data
+        if self.preprocessing_:
+            X = self.preprocessing_.transform(X)
+
+        # Calculate outliers based on samples with too high leverage
+        leverage = calculate_leverage(self.model_, X)
+        return np.where(leverage > self.critical_value_, -1, 1)
+
+    def predict_residuals(
+        self, X: np.ndarray, y: Optional[np.ndarray], validate: bool = True
+    ) -> np.ndarray:
+        """Calculate the leverage of the samples.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        Returns
+        -------
+        np.ndarray
+            Leverage of the samples
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        if validate:
+            X = validate_data(self, X, ensure_2d=True, dtype=np.float64)
+
+        # Apply preprocessing if available
+        if self.preprocessing_:
+            X = self.preprocessing_.transform(X)
+
+        # Calculate the leverage
+        return calculate_leverage(self.model_, X)
+
+    def _calculate_critical_value(self, X: Optional[np.ndarray]) -> float:
+        """Calculate the critical value for outlier detection using the percentile outlier method."""
+
+        # Calculate the leverage of the samples
+        leverage = calculate_leverage(self.model_, X)
+
+        # Calculate the critical value
+        return np.percentile(leverage, self.confidence * 100)
+
+
+def calculate_leverage(model: ModelTypes, X: Optional[np.ndarray]) -> np.ndarray:
+    """
+    Calculate the leverage of the training samples in a PLS/PCA-like model.
+
+    Parameters
+    ----------
+    model : Union[_BasePCA, _PLS]
+        A fitted PCA/PLS model
+
+    X : np.ndarray
+        Preprocessed input data
+
+    Returns
+    -------
+    np.ndarray
+        Leverage of the samples
+    """
+
+    X_transformed = model.transform(X)
+
+    X_hat = (
+        X_transformed @ np.linalg.inv(X_transformed.T @ X_transformed) @ X_transformed.T
+    )
+
+    return np.diag(X_hat)

chemotools/outliers/q_residuals.py

@@ -0,0 +1,225 @@
+from typing import Optional, Literal, Union
+
+import numpy as np
+
+from scipy.stats import norm, chi2
+from sklearn.pipeline import Pipeline
+from sklearn.utils.validation import validate_data, check_is_fitted
+
+from ._base import _ModelResidualsBase, ModelTypes
+
+
+class QResiduals(_ModelResidualsBase):
+    """
+    Calculate Q residuals (Squared Prediction Error - SPE) for PCA or PLS models.
+
+    Parameters
+    ----------
+    model : Union[ModelType, Pipeline]
+        A fitted PCA/PLS model or Pipeline ending with such a model.
+
+    confidence : float, default=0.95
+        Confidence level for statistical calculations (between 0 and 1).
+
+    method : str, default="chi-square"
+        The method used to compute the confidence threshold for Q residuals.
+        Options:
+        - "chi-square" : Uses mean and standard deviation to approximate Q residuals threshold.
+        - "jackson-mudholkar" : Uses eigenvalue-based analytical approximation.
+        - "percentile" : Uses empirical percentile threshold.
+
+    Attributes
+    ----------
+    model_ : ModelType
+        The fitted model of type _BasePCA or _PLS.
+
+    preprocessing_ : Optional[Pipeline]
+        Preprocessing steps before the model.
+
+    n_features_in_ : int
+        Number of features in the input data.
+
+    n_components_ : int
+        Number of components in the model.
+
+    n_samples_ : int
+        Number of samples used to train the model.
+
+    critical_value_ : float
+        The calculated critical value for outlier detection.
+
+    References
+    ----------
+    Johan A. Westerhuis, Stephen P. Gurden, Age K. Smilde (2001) Generalized contribution plots in multivariate statistical process
+    monitoring  Chemometrics and Intelligent Laboratory Systems 51 2000 95–114
+    """
+
+    def __init__(
+        self,
+        model: Union[ModelTypes, Pipeline],
+        confidence: float = 0.95,
+        method: Literal["chi-square", "jackson-mudholkar", "percentile"] = "percentile",
+    ) -> None:
+        self.method = method
+        super().__init__(model, confidence)
+
+    def fit(self, X: np.ndarray, y: Optional[np.ndarray] = None) -> "QResiduals":
+        """
+        Fit the Q Residuals model by computing residuals from the training set.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Training data.
+
+        Returns
+        -------
+        self : object
+            Fitted instance of QResiduals.
+        """
+        X = validate_data(self, X, ensure_2d=True, dtype=np.float64)
+
+        if self.preprocessing_:
+            X = self.preprocessing_.fit_transform(X)
+
+        # Compute the critical threshold using the chosen method
+        self.critical_value_ = self._calculate_critical_value(X)
+
+        return self
+
+    def predict(self, X: np.ndarray) -> np.ndarray:
+        """Identify outliers in the input data based on Q residuals threshold.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data.
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Boolean array indicating outliers (-1 for outliers, 1 for normal data).
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        # Calculate outliers based on the Q residuals
+        Q_residuals = self.predict_residuals(X, validate=False)
+        return np.where(Q_residuals > self.critical_value_, -1, 1)
+
+    def predict_residuals(
+        self, X: np.ndarray, y: Optional[np.ndarray] = None, validate: bool = True
+    ) -> np.ndarray:
+        """Calculate Q residuals (Squared Prediction Error - SPE) for input data.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data.
+
+        validate : bool, default=True
+            Whether to validate the input data.
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Q residuals for each sample.
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        if validate:
+            X = validate_data(self, X, ensure_2d=True, dtype=np.float64)
+
+        # Apply preprocessing if available
+        if self.preprocessing_:
+            X = self.preprocessing_.transform(X)
+
+        # Compute reconstruction error (Q residuals)
+        X_transformed = self.model_.transform(X)
+        X_reconstructed = self.model_.inverse_transform(X_transformed)
+        Q_residuals = np.sum((X - X_reconstructed) ** 2, axis=1)
+
+        return Q_residuals
+
+    def _calculate_critical_value(
+        self,
+        X: Optional[np.ndarray] = None,
+    ) -> float:
+        """Calculate the critical value for outlier detection.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data.
+
+        X_reconstructed : array-like of shape (n_samples, n_features)
+            Reconstructed input data.
+
+        method : str Literal["chi-square", "jackson-mudholkar", "percentile"]
+            The method used to compute the confidence threshold for Q residuals.
+
+        Returns
+        -------
+        float
+            The calculated critical value for outlier detection.
+
+        """
+        # Compute Q residuals for training data
+        X_transformed = self.model_.transform(X)
+        X_reconstructed = self.model_.inverse_transform(X_transformed)
+        residuals = X - X_reconstructed
+
+        if self.method == "chi-square":
+            return self._chi_square_threshold(residuals)
+        elif self.method == "jackson-mudholkar":
+            return self._jackson_mudholkar_threshold(residuals)
+        elif self.method == "percentile":
+            Q_residuals = np.sum((residuals) ** 2, axis=1)
+            return self._percentile_threshold(Q_residuals)
+        else:
+            raise ValueError(
+                "Invalid method. Choose from 'chi-square', 'jackson-mudholkar', or 'percentile'."
+            )
+
+    def _chi_square_threshold(self, residuals: np.ndarray) -> float:
+        """Compute Q residual threshold using Chi-Square Approximation."""
+        eigenvalues = np.linalg.trace(np.cov(residuals.T))
+
+        theta_1 = np.sum(eigenvalues)
+        theta_2 = np.sum(eigenvalues**2)
+        # Degrees of freedom approximation
+        g = theta_2 / theta_1
+        h = (2 * theta_1**2) / theta_2
+
+        # Compute chi-square critical value at given confidence level
+        chi_critical = chi2.ppf(self.confidence, df=h)
+
+        # Compute final Q residual threshold
+        return g * chi_critical
+
+    def _jackson_mudholkar_threshold(self, residuals: np.ndarray) -> float:
+        """Compute Q residual threshold using Jackson & Mudholkar’s analytical method."""
+
+        eigenvalues = np.linalg.trace(np.cov(residuals.T))
+        theta_1 = np.sum(eigenvalues)
+        theta_2 = np.sum(eigenvalues**2)
+        theta_3 = np.sum(eigenvalues**3)
+        z_alpha = norm.ppf(self.confidence)
+
+        h0 = 1 - (2 * theta_1 * theta_3) / (3 * theta_2**2)
+
+        term1 = theta_2 * h0 * (1 - h0) / theta_1**2
+        term2 = np.sqrt(z_alpha * 2 * theta_2 * h0**2) / theta_1
+
+        return theta_1 * (1 - term1 + term2) ** (1 / h0)
+
+    def _percentile_threshold(self, Q_residuals: np.ndarray) -> float:
+        """Compute Q residual threshold using the empirical percentile method."""
+        return np.percentile(Q_residuals, self.confidence * 100)

chemotools/outliers/studentized_residuals.py

@@ -0,0 +1,197 @@
+from typing import Optional, Union
+import numpy as np
+
+from sklearn.cross_decomposition._pls import _PLS
+from sklearn.pipeline import Pipeline
+from sklearn.utils.validation import validate_data, check_is_fitted
+
+
+from ._base import _ModelResidualsBase, ModelTypes
+from .leverage import calculate_leverage
+
+
+class StudentizedResiduals(_ModelResidualsBase):
+    """
+    Calculate the Studentized Residuals on a _PLS model preditions.
+
+    Parameters
+    ----------
+    model : Union[ModelType, Pipeline]
+        A fitted _PLS model or Pipeline ending with such a model
+
+    Attributes
+    ----------
+    model_ : ModelType
+        The fitted model of type _BasePCA or _PLS
+
+    preprocessing_ : Optional[Pipeline]
+        Preprocessing steps before the model
+
+    References
+    ----------
+
+    """
+
+    def __init__(self, model: Union[_PLS, Pipeline], confidence=0.95) -> None:
+        super().__init__(model, confidence)
+
+    def fit(self, X: np.ndarray, y: Optional[np.ndarray]) -> "StudentizedResiduals":
+        """
+        Fit the model to the input data.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        y : array-like of shape (n_samples,)
+            Target data
+        """
+        # Validate the input data
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        # Preprocess the data
+        if self.preprocessing_:
+            X = self.preprocessing_.transform(X)
+
+        # Calculate y residuals
+        y_residuals = y - self.model_.predict(X)
+        y_residuals = (
+            y_residuals.reshape(-1, 1) if len(y_residuals.shape) == 1 else y_residuals
+        )
+
+        # Calculate the studentized residuals
+        studentized_residuals = calculate_studentized_residuals(
+            self.model_, X, y_residuals
+        )
+
+        # Calculate the critical threshold
+        self.critical_value_ = self._calculate_critical_value(studentized_residuals)
+
+        return self
+
+    def predict(self, X: np.ndarray, y: Optional[np.ndarray]) -> np.ndarray:
+        """Calculate studentized residuals in the model predictions. and return a boolean array indicating outliers.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        y : array-like of shape (n_samples,)
+            Target data
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Studentized residuals of the predictions
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        X = validate_data(
+            self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
+        )
+
+        # Preprocess the data
+        if self.preprocessing_:
+            X = self.preprocessing_.transform(X)
+
+        # Calculate y residuals
+        y_residuals = y - self.model_.predict(X)
+        y_residuals = (
+            y_residuals.reshape(-1, 1) if len(y_residuals.shape) == 1 else y_residuals
+        )
+
+        # Calculate the studentized residuals
+        studentized_residuals = calculate_studentized_residuals(
+            self.model_, X, y_residuals
+        )
+        return np.where(studentized_residuals > self.critical_value_, -1, 1)
+
+    def predict_residuals(
+        self, X: np.ndarray, y: Optional[np.ndarray], validate: bool = True
+    ) -> np.ndarray:
+        """Calculate the studentized residuals of the model predictions.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Input data
+
+        y : array-like of shape (n_samples,)
+            Target values
+
+        Returns
+        -------
+        ndarray of shape (n_samples,)
+            Studentized residuals of the model predictions
+        """
+        # Check the estimator has been fitted
+        check_is_fitted(self, ["critical_value_"])
+
+        # Validate the input data
+        if validate:
+            X = validate_data(self, X, ensure_2d=True, dtype=np.float64)
+
+        # Apply preprocessing if available
+        if self.preprocessing_:
+            X = self.preprocessing_.transform(X)
+
+        # Calculate y residuals
+        y_residuals = y - self.model_.predict(X)
+        y_residuals = (
+            y_residuals.reshape(-1, 1) if len(y_residuals.shape) == 1 else y_residuals
+        )
+
+        return calculate_studentized_residuals(self.model_, X, y_residuals)
+
+    def _calculate_critical_value(self, X: Optional[np.ndarray]) -> float:
+        """Calculate the critical value for outlier detection.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples,)
+            Studentized residuals
+
+        Returns
+        -------
+        float
+            The calculated critical value for outlier detection
+        """
+
+        return np.percentile(X, self.confidence * 100) if X is not None else 0.0
+
+
+def calculate_studentized_residuals(
+    model: ModelTypes, X: np.ndarray, y_residuals: np.ndarray
+) -> np.ndarray:
+    """Calculate the studentized residuals of the model predictions.
+
+    Parameters
+    ----------
+    model : ModelTypes
+        A fitted model
+
+    X : array-like of shape (n_samples, n_features)
+        Input data
+
+    y : array-like of shape (n_samples,)
+        Target values
+
+    Returns
+    -------
+    ndarray of shape (n_samples,)
+        Studentized residuals of the model predictions
+    """
+
+    # Calculate the leverage of the samples
+    leverage = calculate_leverage(model, X)
+
+    # Calculate the standard deviation of the residuals
+    std = np.sqrt(np.sum(y_residuals**2, axis=0) / (X.shape[0] - model.n_components))
+
+    return (y_residuals / (std * np.sqrt(1 - leverage.reshape(-1, 1)))).flatten()

{chemotools-0.1.9.dist-info → chemotools-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: chemotools
-Version: 0.1.9
+Version: 0.1.10
 Summary: chemotools: A Python Package that Integrates Chemometrics and scikit-learn
 License: MIT
 Author: Pau Cabaneros

{chemotools-0.1.9.dist-info → chemotools-0.1.10.dist-info}/RECORD

@@ -1,8 +1,9 @@
 chemotools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chemotools/augmentation/__init__.py,sha256=iRltqvskLJAhwxxlTtGPWJfR5XiwvQfqbv4QdUMF9BU,318
+chemotools/augmentation/__init__.py,sha256=ohlRHgRWTkvNpO3RikKtowzboqunQqx0WqtNccuWOHw,397
 chemotools/augmentation/_add_noise.py,sha256=fkTJfIYtZXezcjy6Vz8asIhpBoVp4oaIifppK9vZpM8,4362
 chemotools/augmentation/_baseline_shift.py,sha256=kIlYvmKS9pu9vh_-eZ7PSHPuH_58V9mgYbSJt6Gq3BA,3476
 chemotools/augmentation/_fractional_shift.py,sha256=dJ0Vuc-U02HhjKkOwc48qnOksZYgbHwL2ko7tWCZTQU,6916
+chemotools/augmentation/_gaussian_broadening.py,sha256=dJsPlTKqpecKaCDU3vOvedIb-t_HyCkQprxNv0DmYZQ,4236
 chemotools/augmentation/_index_shift.py,sha256=BTtadweDvvMtiF8t7ldwsE6Kl6FmKLCkVJjSzSWyIDs,6904
 chemotools/augmentation/_spectrum_scale.py,sha256=hMsmzXpssbI7tGm_YnQn9wjbByso3CgVxd3Hs8kfLS8,3442
 chemotools/baseline/__init__.py,sha256=VzoblGg8Hx_FkTc_n7a-ZjGvtKP8JE_NwJKWenGFQkM,584
@@ -29,6 +30,14 @@ chemotools/derivative/_savitzky_golay.py,sha256=CuCrKoLmrB1YmJ4ihIykgkL3tO3frqkS
 chemotools/feature_selection/__init__.py,sha256=1_i28hIxijjwhMypTy1w2fLbzXXVkKD5IYzzY8ZSuHw,117
 chemotools/feature_selection/_index_selector.py,sha256=lNTP2b7P3doWl30KiAr3Xd2HOMxeUmj24MuqoXl4Voc,3556
 chemotools/feature_selection/_range_cut.py,sha256=lVVVC30ZsK2z9jsDGb_z6l8Ty2I89yM05_dIDbMP73Q,3564
+chemotools/outliers/__init__.py,sha256=wpdlyqU34n1Pb9kGCM4idhcok35WAakxEhzP0xeKaZw,272
+chemotools/outliers/_base.py,sha256=zx9z_Snkvq5YWBoRi_-kRr3a-Q7jTz1zVlrGWycUTb4,5232
+chemotools/outliers/_utils.py,sha256=SDrrDvgEVQyPuKdh0Rw0DD4a8LShbNAQLRwSLICtiYU,2720
+chemotools/outliers/dmodx.py,sha256=R9LaQpUJeDv4GJ0hroKOlFcFbsfQRtrHWD_EI3-TX7Y,4521
+chemotools/outliers/hotelling_t2.py,sha256=Ga1qmlurF_fps9igaTUGOrnUOctIJEYqoCdb468KhY4,5006
+chemotools/outliers/leverage.py,sha256=zgxG2F7ZCf5wRVJeezHSJ2gaUDTP2CvK38Rr-hR6niA,4215
+chemotools/outliers/q_residuals.py,sha256=6_h73A1YxHBcQtjAXOAp1Rb7egHJwj0DQ0MKdnj6aBQ,7647
+chemotools/outliers/studentized_residuals.py,sha256=rF0wObKQV5DCa8THkZcuwdu7u4mBk-dbOHth5tj5cqM,5830
 chemotools/scale/__init__.py,sha256=eztqcHg-TKE1Rr0N9ArfytHk8teuqVfi4SZi2DS96vc,175
 chemotools/scale/_min_max_scaler.py,sha256=YvqRkV2pXu-viQrpjzWcp9KmSSCYSoubSnrZHRLqgKQ,3011
 chemotools/scale/_norm_scaler.py,sha256=CHWSir2q-pL1hxzw_ZB45yi4mw-SkJ4YOa1CUL4nm2I,2568
@@ -44,7 +53,7 @@ chemotools/smooth/_median_filter.py,sha256=9ndTJCwrZirWlvDNldiigMddy79KIGq9OwwYN
 chemotools/smooth/_savitzky_golay_filter.py,sha256=27iFUWxdL9_7oZabR0R5L0ZTpBmYfVUjx2XCTukihBE,3509
 chemotools/smooth/_whittaker_smooth.py,sha256=lpLAyf4GdyDW4ulT1nyEoK6xQEl2cVUKquawQdGWbHU,3571
 chemotools/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chemotools-0.1.9.dist-info/LICENSE,sha256=qtyOy2wDQVX9hxp58h3T-6Lmfv-mSCHoSRkcLUdM9bg,1070
-chemotools-0.1.9.dist-info/METADATA,sha256=9sih25qSOTJX36gib96hxm-e86TorD040r_WNFzbc9U,5239
-chemotools-0.1.9.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-chemotools-0.1.9.dist-info/RECORD,,
+chemotools-0.1.10.dist-info/LICENSE,sha256=qtyOy2wDQVX9hxp58h3T-6Lmfv-mSCHoSRkcLUdM9bg,1070
+chemotools-0.1.10.dist-info/METADATA,sha256=fRgOO8cS2JNtNWz_CEG0uKvncSHEJ8myfhm2IOz3y-4,5240
+chemotools-0.1.10.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+chemotools-0.1.10.dist-info/RECORD,,

{chemotools-0.1.9.dist-info → chemotools-0.1.10.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.0.1
+Generator: poetry-core 2.1.1
 Root-Is-Purelib: true
 Tag: py3-none-any