esem.utils.get_param_mask¶
- esem.utils.get_param_mask(X, y, criterion='bic', **kwargs)¶
Determine the most relevant parameters in the input space using a regularised linear model and either the Aikake or Baysian Information Criterion.
- Parameters
X (array-like of
shape (n_samples
,n_features)
) – Parameter valuesy (array-like of
shape (n_samples,)
) – target values.criterion (
{'bic', 'aic'}
, default'bic'
) – The information criteria to apply for parameter selection. Either Aikake or Baysian Information Criterion.kwargs (
dict
) – Further arguments for sklearn.feature_selection.SelectFromModel
- Returns
mask (
ndarray
) – A boolean array of shape [# input features], in which an element is True iff its corresponding feature is selected for retention.