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中。避免類型問題。