重复编辑最近邻#

class imblearn.under_sampling.RepeatedEditedNearestNeighbours(*, sampling_strategy='auto', n_neighbors=3, max_iter=100, kind_sel='all', n_jobs=None)[source]#

基于重复编辑最近邻方法进行欠采样。

此方法重复多次EditedNearestNeighbours算法。 重复将在以下情况下停止:i) 达到最大迭代次数, 或 ii) 不再有观测值被移除,或 iii) 其中一个多数类 变为少数类,或 iv) 在欠采样过程中其中一个多数类消失。

更多内容请参阅用户指南

Parameters:
sampling_strategystr, list or callable

采样信息以对数据集进行采样。

  • str 时,指定重采样所针对的类别。请注意,每个类别的样本数量不会相等。可能的选择有:

    'majority': 仅对多数类进行重采样;

    'not minority': 对除少数类之外的所有类进行重采样;

    'not majority': 重新采样除多数类之外的所有类;

    'all': 对所有类别进行重采样;

    'auto': 等同于 'not minority'.

  • list时,列表包含重采样所针对的类。

  • 当可调用时,函数接受 y 并返回一个 dict。键对应于目标类别。值对应于每个类别所需的样本数量。

n_neighborsint or object, default=3

如果 int,考虑用于欠采样的邻域大小,即, 如果 n_neighbors=3,当样本的3个最近邻居中的任何一个或大多数来自不同类别时,该样本将被移除。如果是对象,则使用继承自 KNeighborsMixin 的估计器来查找最近邻居。请注意,如果您想检查样本的3个最近邻居以进行欠采样,您需要传递一个4-KNN。

max_iterint, default=100

编辑最近邻的最大迭代次数。

kind_sel{‘all’, ‘mode’}, default=’all’

用于排除样本的策略。

  • 如果 'all',所有邻居应与被检查样本属于同一类别,否则将被排除。

  • 如果 'mode',大多数邻居应该与被检查样本的类别相同,以便不被排除。

策略 "all" 将比 'mode' 更不保守。因此,当 kind_sel="all" 时,通常会移除更多的样本。

n_jobsint, default=None

在交叉验证循环中使用的CPU核心数量。 None 表示1,除非在 joblib.parallel_backend 上下文中。 -1 表示使用所有处理器。更多详情请参见 术语表

Attributes:
sampling_strategy_dict

包含用于采样数据集的信息的字典。键对应于要从中采样的类标签,值是要采样的样本数量。

nn_estimator object

已验证的K近邻估计器与参数n_neighbors相关联。

enn_sampler object

已验证的 EditedNearestNeighbours 实例。

sample_indices_ndarray of shape (n_new_samples,)

所选样本的索引。

在版本0.4中添加。

n_iter_int

运行的迭代次数。

在版本0.6中添加。

n_features_in_int

输入数据集中的特征数量。

在版本0.9中添加。

feature_names_in_ndarray of shape (n_features_in_,)

fit期间看到的特征名称。仅在X具有全部为字符串的特征名称时定义。

在版本0.10中添加。

另请参阅

CondensedNearestNeighbour

通过压缩样本进行欠采样。

EditedNearestNeighbours

通过编辑样本进行欠采样。

AllKNN

使用不同邻居的ENN进行欠采样。

注释

该方法基于[1]。在采样一个类时,使用了[1]中提出的一对多方案。

支持多类重采样。

参考文献

[1] (1,2)

I. Tomek, “编辑最近邻规则的实验,” IEEE 系统、人与控制论汇刊, 卷 6(6), 页 448-452, 1976年6月.

示例

>>> from collections import Counter
>>> from sklearn.datasets import make_classification
>>> from imblearn.under_sampling import RepeatedEditedNearestNeighbours
>>> X, y = make_classification(n_classes=2, class_sep=2,
... weights=[0.1, 0.9], n_informative=3, n_redundant=1, flip_y=0,
... n_features=20, n_clusters_per_class=1, n_samples=1000, random_state=10)
>>> print('Original dataset shape %s' % Counter(y))
Original dataset shape Counter({1: 900, 0: 100})
>>> renn = RepeatedEditedNearestNeighbours()
>>> X_res, y_res = renn.fit_resample(X, y)
>>> print('Resampled dataset shape %s' % Counter(y_res))
Resampled dataset shape Counter({1: 887, 0: 100})

方法

fit(X, y, **params)

检查采样器的输入和统计信息。

fit_resample(X, y, **params)

重新采样数据集。

get_feature_names_out([input_features])

获取转换的输出特征名称。

get_metadata_routing()

获取此对象的元数据路由。

get_params([deep])

获取此估计器的参数。

set_params(**params)

设置此估计器的参数。

fit(X, y, **params)[source]#

检查采样器的输入和统计信息。

在所有情况下,您都应该使用 fit_resample

Parameters:
X{array-like, dataframe, sparse matrix} of shape (n_samples, n_features)

数据数组。

yarray-like of shape (n_samples,)

目标数组。

Returns:
selfobject

返回实例本身。

fit_resample(X, y, **params)[source]#

重新采样数据集。

Parameters:
X{array-like, dataframe, sparse matrix} of shape (n_samples, n_features)

包含需要采样的数据的矩阵。

yarray-like of shape (n_samples,)

X中每个样本对应的标签。

Returns:
X_resampled{array-like, dataframe, sparse matrix} of shape (n_samples_new, n_features)

包含重采样数据的数组。

y_resampledarray-like of shape (n_samples_new,)

X_resampled 对应的标签。

get_feature_names_out(input_features=None)[source]#

获取转换的输出特征名称。

Parameters:
input_featuresarray-like of str or None, default=None

输入特征。

  • 如果 input_featuresNone,则使用 feature_names_in_ 作为特征名称。如果 feature_names_in_ 未定义,则生成以下输入特征名称: ["x0", "x1", ..., "x(n_features_in_ - 1)"]

  • 如果 input_features 是类似数组的,那么 input_features 必须 与 feature_names_in_ 匹配,如果 feature_names_in_ 已定义。

Returns:
feature_names_outndarray of str objects

与输入特征相同。

get_metadata_routing()[source]#

获取此对象的元数据路由。

请查看用户指南了解路由机制的工作原理。

Returns:
routingMetadataRequest

一个封装路由信息的MetadataRequest

get_params(deep=True)[source]#

获取此估计器的参数。

Parameters:
deepbool, default=True

如果为True,将返回此估计器及其包含的子对象的参数。

Returns:
paramsdict

参数名称映射到它们的值。

set_params(**params)[source]#

设置此估计器的参数。

该方法适用于简单的估计器以及嵌套对象(如Pipeline)。后者具有__形式的参数,以便可以更新嵌套对象的每个组件。

Parameters:
**paramsdict

估计器参数。

Returns:
selfestimator instance

估计器实例。

使用imblearn.under_sampling.RepeatedEditedNearestNeighbours的示例#

比较欠采样采样器

Compare under-sampling samplers