Source code for chemotools.smooth._mean_filter
"""
The :mod:`chemotools.smooth._mean_filter` module implements the Mean Filter (MF) transformation.
"""
# Authors: Pau Cabaneros
# License: MIT
import numpy as np
from scipy.ndimage import uniform_filter1d
from sklearn.base import BaseEstimator, TransformerMixin, OneToOneFeatureMixin
from sklearn.utils.validation import check_is_fitted, validate_data
[docs]
class MeanFilter(TransformerMixin, OneToOneFeatureMixin, BaseEstimator):
"""
A transformer that calculates the mean filter of the input data.
Parameters
----------
window_size : int, optional, default=3
The size of the window to use for the mean filter. Must be odd. Default is 3.
mode : str, optional, default="nearest"
The mode to use for the mean filter. Can be "nearest", "constant", "reflect",
"wrap", "mirror" or "interp". Default is "nearest".
Attributes
----------
n_features_in_ : int
The number of features in the training data.
Examples
--------
>>> from chemotools.datasets import load_fermentation_train
>>> from chemotools.smooth import MeanFilter
>>> # Load sample data
>>> X, _ = load_fermentation_train()
>>> # Initialize MeanFilter
>>> mf = MeanFilter()
MeanFilter()
>>> # Fit and transform the data
>>> X_smoothed = mf.fit_transform(X)
"""
def __init__(self, window_size: int = 3, mode="nearest") -> None:
self.window_size = window_size
self.mode = mode
[docs]
def fit(self, X: np.ndarray, y=None) -> "MeanFilter":
"""
Fit the transformer to the input data.
Parameters
----------
X : np.ndarray of shape (n_samples, n_features)
The input data to fit the transformer to.
y : None
Ignored to align with API.
Returns
-------
self : MeanFilter
The fitted transformer.
"""
# Check that X is a 2D array and has only finite values
X = validate_data(
self, X, y="no_validation", ensure_2d=True, reset=True, dtype=np.float64
)
return self
