Source code for tropea_clustering._internal.onion_uni

"""onion-clustering for univariate time-series."""

# Author: Becchi Matteo <bechmath@gmail.com>
# Reference: https://www.pnas.org/doi/abs/10.1073/pnas.2403771121

from typing import Union

import numpy as np
from sklearn.base import BaseEstimator, ClusterMixin

from tropea_clustering._internal.main import _main as _onion_inner


[docs] def onion_uni( X: np.ndarray, bins: Union[str, int] = "auto", number_of_sigmas: float = 2.0, ): """ Performs onion clustering on the data array 'X'. Parameters ---------- X : ndarray of shape (n_particles * n_windows, tau_window) The raw data. Notice that each signal window is considered as a single data point. bins : int, default="auto" The number of bins used for the construction of the histograms. Can be an integer value, or "auto". If "auto", the default of numpy.histogram_bin_edges is used (see https://numpy.org/doc/stable/reference/generated/numpy.histogram_bin_edges.html#numpy.histogram_bin_edges). number_of_sigmas : float, default=2.0 Sets the thresholds for classifing a signal window inside a state: the window is contained in the state if it is entirely contained inside number_of_sigmas * state.sigmas times from state.mean. Returns ------- states_list : List[StateUni] The list of the identified states. Refer to the documentation of StateUni for accessing the information on the states. labels : ndarray of shape (n_particles * n_windows,) Cluster labels for each signal window. Unclassified points are given the label -1. """ est = OnionUni( bins=bins, number_of_sigmas=number_of_sigmas, ) est.fit(X) return est.state_list_, est.labels_
[docs] class OnionUni(BaseEstimator, ClusterMixin): """ Performs onion clustering on a data array. Parameters ---------- bins : int, default="auto" The number of bins used for the construction of the histograms. Can be an integer value, or "auto". If "auto", the default of numpy.histogram_bin_edges is used (see https://numpy.org/doc/stable/reference/generated/numpy.histogram_bin_edges.html#numpy.histogram_bin_edges). number_of_sigmas : float, default=2.0 Sets the thresholds for classifing a signal window inside a state: the window is contained in the state if it is entirely contained inside number_of_sigmas * state.sigmas times from state.mean. Attributes ---------- state_list_ : List[StateUni] List of the identified states. Refer to the documentation of StateUni for accessing the information on the states. labels_: ndarray of shape (n_particles * n_windows,) Cluster labels for signal window. Unclassified points are given the label -1. """ def __init__( self, bins: Union[str, int] = "auto", number_of_sigmas: float = 2.0, ): self.bins = bins self.number_of_sigmas = number_of_sigmas
[docs] def fit(self, X, y=None): """Performs onion clustering on the data array 'X'. Parameters ---------- X : ndarray of shape (n_particles * n_windows, tau_window) The raw data. Notice that each signal window is considered as a single data point. Returns ------- self : object A fitted instance of self. """ X = self._validate_data(X, accept_sparse=False) if X.ndim != 2: raise ValueError("Expected 2-dimensional input data.") if X.shape[0] <= 1: raise ValueError("n_samples = 1") if X.shape[1] <= 1: raise ValueError("n_features = 1") # Check for complex input if not ( np.issubdtype(X.dtype, np.floating) or np.issubdtype(X.dtype, np.integer) ): raise ValueError("Complex data not supported") X = X.copy() # copy to avoid in-place modification cl_ob = _onion_inner( X, self.bins, self.number_of_sigmas, ) self.state_list_ = cl_ob.state_list self.labels_ = cl_ob.data.labels return self
[docs] def fit_predict(self, X, y=None): """Computes clusters on the data array 'X' and returns labels. Parameters ---------- X : ndarray of shape (n_particles * n_windows, tau_window) The raw data. Notice that each signal window is considered as a single data point. Returns ------- labels_: ndarray of shape (n_particles * n_windows,) Cluster labels for signal window. Unclassified points are given the label -1. """ return self.fit(X).labels_
[docs] def get_params(self, deep=True): return { "bins": self.bins, "number_of_sigmas": self.number_of_sigmas, }
[docs] def set_params(self, **params): for param, value in params.items(): setattr(self, param, value) return self