LeavePGroupsOut#

class sklearn.model_selection.LeavePGroupsOut(n_groups)[源代码]#

Leave P Group(s) Out 交叉验证器。

提供训练/测试索引,以根据第三方提供的组来划分数据。此组信息可用于将样本编码为整数,实现任意域特定的分层。

例如,组可以是样本的采集年份,从而允许针对基于时间的拆分进行交叉验证。

LeavePGroupsOut 和 LeaveOneGroupOut 的区别在于,前者将测试集构建为包含 p 个不同组值的样本,而后者则使用所有分配给同一组的样本。

用户指南 中阅读更多内容。

参数:
n_groupsint

在测试集中留出的组数(p)。

另请参阅

GroupKFold

具有非重叠组的 K 折迭代器变体。

示例

>>> import numpy as np
>>> from sklearn.model_selection import LeavePGroupsOut
>>> X = np.array([[1, 2], [3, 4], [5, 6]])
>>> y = np.array([1, 2, 1])
>>> groups = np.array([1, 2, 3])
>>> lpgo = LeavePGroupsOut(n_groups=2)
>>> lpgo.get_n_splits(groups=groups)
3
>>> print(lpgo)
LeavePGroupsOut(n_groups=2)
>>> for i, (train_index, test_index) in enumerate(lpgo.split(X, y, groups)):
...     print(f"Fold {i}:")
...     print(f"  Train: index={train_index}, group={groups[train_index]}")
...     print(f"  Test:  index={test_index}, group={groups[test_index]}")
Fold 0:
  Train: index=[2], group=[3]
  Test:  index=[0 1], group=[1 2]
Fold 1:
  Train: index=[1], group=[2]
  Test:  index=[0 2], group=[1 3]
Fold 2:
  Train: index=[0], group=[1]
  Test:  index=[1 2], group=[2 3]
get_metadata_routing()[源代码]#

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

请查阅 用户指南,了解路由机制如何工作。

返回:
routingMetadataRequest

封装路由信息的 MetadataRequest

get_n_splits(X=None, y=None, groups=None)[源代码]#

返回交叉验证器中的拆分迭代次数。

参数:
Xarray-like of shape (n_samples, n_features), default=None

始终忽略,存在是为了保持 API 兼容性。

yshape 为 (n_samples,), default=None 的 array-like

始终忽略,存在是为了保持 API 兼容性。

groups形状为 (n_samples,) 的类数组对象,默认=None

在将数据集划分为训练/测试集时使用的样本的组标签。为了计算分割的数量,必须始终指定此“groups”参数,尽管其他参数可以省略。

返回:
n_splitsint

返回交叉验证器中的拆分迭代次数。

set_split_request(*, groups: bool | None | str = '$UNCHANGED$') LeavePGroupsOut[源代码]#

配置是否应请求元数据传递给 split 方法。

请注意,此方法仅在以下情况下相关:此估计器用作 元估计器 中的子估计器,并且通过 enable_metadata_routing=True 启用了元数据路由(请参阅 sklearn.set_config)。请查看 用户指南 以了解路由机制的工作原理。

每个参数的选项如下:

  • True:请求元数据,如果提供则传递给 split。如果未提供元数据,则忽略该请求。

  • False:不请求元数据,元估计器不会将其传递给 split

  • None:不请求元数据,如果用户提供元数据,元估计器将引发错误。

  • str:应将元数据以给定别名而不是原始名称传递给元估计器。

默认值 (sklearn.utils.metadata_routing.UNCHANGED) 保留现有请求。这允许您更改某些参数的请求而不更改其他参数。

在版本 1.3 中新增。

参数:
groupsstr, True, False, or None, default=sklearn.utils.metadata_routing.UNCHANGED

splitgroups 参数的元数据路由。

返回:
selfobject

更新后的对象。

split(X, y=None, groups=None)[源代码]#

生成索引以将数据拆分为训练集和测试集。

参数:
Xshape 为 (n_samples, n_features) 的 array-like

训练数据,其中 n_samples 是样本数,n_features 是特征数。

yshape 为 (n_samples,), default=None 的 array-like

用于监督学习问题的目标变量。

groupsarray-like of shape (n_samples,)

用于将数据集拆分为训练/测试集时样本的组标签。

生成:
trainndarray

该拆分的训练集索引。

testndarray

该拆分的测试集索引。