multivelo.gen_wnn

multivelo.gen_wnn(adata_rna, adata_adt, dims, nn, random_state=0)

Computes inputs for KNN smoothing.

This function calculates the nn_idx and nn_dist matrices needed to run knn_smooth_chrom().

Parameters:

• adata_rna (AnnData) – RNA anndata object.

• adata_atac (AnnData) – ATAC anndata object.

• dims (List[int]) – Dimensions of data for RNA (index=0) and ATAC (index=1)

• nn (int (default: None)) – Top N neighbors to extract for each cell in the connectivities matrix.

Returns:

• nn_idx (np.darray (default: None)) – KNN index matrix of size (cells, k).

• nn_dist (np.darray (default: None)) – KNN distance matrix of size (cells, k).