列或一维数组#
- 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])