ITMO_FS.filters.multivariate.STIR

class ITMO_FS.filters.multivariate.STIR(n_features, metric='manhattan', k=1)

Feature selection using STIR algorithm.

Parameters:
  • n_features (int) – Number of features to select.
  • metric (str or callable) – Distance metric to use in kNN. If str, should be one of the standard distance metrics (e.g. ‘euclidean’ or ‘manhattan’). If callable, should have the signature metric(x1 (array-like, shape (n,)), x2 (array-like, shape (n,))) that should return the distance between two vectors.
  • k (int) – Number of constant nearest hits/misses.

Notes

For more details see this paper.

Examples

>>> from ITMO_FS.filters.multivariate import STIR
>>> import numpy as np
>>> X = np.array([[3, 3, 3, 2, 2], [3, 3, 1, 2, 3], [1, 3, 5, 1, 1],
... [3, 1, 4, 3, 1], [3, 1, 2, 3, 1]])
>>> y = np.array([1, 2, 2, 1, 2])
>>> model = STIR(2).fit(X, y)
>>> model.selected_features_
array([2, 0], dtype=int64)
__init__(n_features, metric='manhattan', k=1)

Initialize self. See help(type(self)) for accurate signature.

fit(X, y=None, **fit_params)

Fit the algorithm.

Parameters:
  • X (array-like, shape (n_samples, n_features)) – The training input samples.
  • y (array-like, shape (n_samples,), optional) – The class labels.
  • fit_params (dict, optional) – Additional parameters to pass to underlying _fit function.
Returns:
Return type:

Self, i.e. the transformer object.
fit_transform(X, y=None, **fit_params)

Fit to data, then transform it.

Fits transformer to X and y with optional parameters fit_params and returns a transformed version of X.

Parameters:
  • X ({array-like, sparse matrix, dataframe} of shape (n_samples, n_features)) –
  • y (ndarray of shape (n_samples,), default=None) – Target values.
  • **fit_params (dict) – Additional fit parameters.
Returns:

X_new – Transformed array.
Return type:

ndarray array of shape (n_samples, n_features_new)
get_params(deep=True)

Get parameters for this estimator.

Parameters:deep (bool, default=True) – If True, will return the parameters for this estimator and contained subobjects that are estimators.
Returns:params – Parameter names mapped to their values.
Return type:mapping of string to any
set_params(**params)

Set the parameters of this estimator.

The method works on simple estimators as well as on nested objects (such as pipelines). The latter have parameters of the form <component>__<parameter> so that it’s possible to update each component of a nested object.

Parameters:**params (dict) – Estimator parameters.
Returns:self – Estimator instance.
Return type:object
transform(X)

Transform given data by slicing it with selected features.

Parameters:X (array-like, shape (n_samples, n_features)) – The training input samples.
Returns:
Return type:Transformed 2D numpy array