column_or_1d#
- sklearn.utils.validation.column_or_1d(y, *, dtype=None, warn=False, device=None)[源]#
展平列或一维 NumPy 数组,否则会引发错误。
- 参数:
- y类数组
输入数据。
- dtype数据类型, 默认为None
y
的数据类型。在 1.2 版本中新增。
- warn布尔值, 默认为False
控制警告的显示。
- device设备, 默认为None
device
对象。更多详情请参见 Array API 用户指南。在 1.6 版本中新增。
- 返回:
- yndarray
输出数据。
- 引发:
- ValueError
如果
y
不是一维数组,也不是单行或单列的二维数组。
示例
>>> from sklearn.utils.validation import column_or_1d >>> column_or_1d([1, 1]) array([1, 1])