geneGATer.pl.attention_umap#

geneGATer.pl.attention_umap(model, data, adata, n_components=10, random_state=1337, font_size=14, cmap='tab20', cluster_key='cluster', umap_model='umap', set_size=(10, 10))#

Create Attention UMAP plot.

Parameters

  • model – The model from learn_model function.

  • data – The data from learn_model function.

  • adata (AnnData) – The AnnData object.

  • n_components (default: 10) – Number of components for the PCA.

  • random_state (default: 1337) – Random state for reproducibility.

  • font_size (default: 14) – Font size for the plot.

  • cmap (default: 'tab20') – Color map for the plot.

  • cluster_key (default: 'cluster') – Key in which clustering is getting saved in adata.obs.

  • umap_model (default: 'umap') – “umap” or “tsne”.

  • set_size (default: (10, 10)) – Size of the plot.

Returns

Attention UMAP plot.

-fig (figure)

The matplotlib figure object.

-umap (ndarray)

The UMAP coordinates.