chemotools 0.1.9__tar.gz → 0.1.10__tar.gz

{chemotools-0.1.9 → chemotools-0.1.10}/PKG-INFO

@@ -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 → chemotools-0.1.10}/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-0.1.10/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-0.1.10/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-0.1.10/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-0.1.10/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-0.1.10/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)