indexable#

sklearn.utils.indexable(*iterables)[source]#

使数组可用于交叉验证的索引。

检查长度是否一致,透传 None,并通过将稀疏矩阵转换为 csr 格式以及将非可迭代对象转换为数组,确保所有内容均可被索引。

参数:
*iterables{列表, 数据框, ndarray, 稀疏矩阵}

需要确保可切片性的对象列表。

返回:
result{ndarray, 稀疏矩阵, 数据框} 列表或 None

返回一个包含可索引数组(即 NumPy 数组、稀疏矩阵或数据框)的列表,或返回 None

示例

>>> from sklearn.utils import indexable
>>> from scipy.sparse import csr_array
>>> import numpy as np
>>> iterables = [
...     [1, 2, 3], np.array([2, 3, 4]), None, csr_array([[5], [6], [7]])
... ]
>>> indexable(*iterables)
[[1, 2, 3], array([2, 3, 4]), None, <...Sparse...dtype 'int64'...shape (3, 1)>]