geneGATer.pl.attention_matrix#

geneGATer.pl.attention_matrix(model, data, font_size=14, cmap='Reds')#

Create Attention Matrix plot.

Parameters

  • model – The model from learn_model function.

  • data – The data from learn_model function.

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

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

Returns

Attention Matrix plot.

-fig (figure)

The matplotlib figure object.

-adj (csr_matrix) The attention matrix.