列或一维数组#

sklearn.utils.validation.column_or_1d(y, *, dtype=None, warn=False, device=None)[source]#

展平列向量或一维 NumPy 数组,否则引发错误。

参数:
y类数组

输入数据。

dtype数据类型,默认=None

y的数据类型。

1.2 版本新增。

warn布尔值,默认=False

控制警告的显示。

device设备,默认=None

device 对象。更多详情请参见 数组 API 用户指南

1.6 版本新增。

返回:
yndarray

输出数据。

引发:
ValueError

如果y不是一维数组或只有一行或一列的二维数组。

示例

>>> from sklearn.utils.validation import column_or_1d
>>> column_or_1d([1, 1])
array([1, 1])