scSLAT.model.graphmodel.LGCN

class scSLAT.model.graphmodel.LGCN(input_size, K=8)

Bases: Module

Lightweight GCN which remove nonlinear functions and concatenate the embeddings of each layer:

(\(Z = f_{e}(A, X) = Concat( [X, A_{X}, A_{2X}, ..., A_{KX}])W_{e}\))

Parameters:

• input_size (int) – input dim

• K (typing.Optional[int]) – LGCN layers

Methods

forward

Defines the computation performed at every call.

Attributes