sklearn.feature_extraction.grid_to_graph

sklearn.feature_extraction.image.grid_to_graph(n_x, n_y, n_z=1, *, mask=None, return_as=<class 'scipy.sparse.coo.coo_matrix'>, dtype=<class 'int'>)

[源码]

像素到像素的连接图.

如果两个体素相连,则边就存在。

参数 说明
n_x int
x轴尺寸
n_y int
y轴尺寸
n_z int, default=1
z轴尺寸
mask ndarray of shape (n_x, n_y, n_z), dtype=bool, default=None
可选的图像遮罩,只考虑部分像素。
return_as np.ndarray or a sparse matrix class, default=sparse.coo_matrix
用来构建返回邻接矩阵的类。
dtype dtype, default=int
返回稀疏矩阵的数据。默认情况下,它是int

注意

对于scikit-learn版本0.14.1和之前版本,return_as=np。ndarray是通过返回稠密np来处理的。矩阵实例。展望未来,np。ndarray返回一个np。ndarray,如预期。

为了兼容性,依赖于此方法的用户代码应该将其调用封装在np中。避免类型问题。