scSLAT.model.graphmodel.LGCN_mlp

class scSLAT.model.graphmodel.LGCN_mlp(input_size, output_size, K=8, hidden_size=512, dropout=0.2)

Bases: Module

Add one hidden layer MLP in LGCN

Parameters:

• input_size (int) – input dim

• output_size (int) – output dim

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

• hidden_size (typing.Optional[int]) – hidden size of MLP

• dropout (typing.Optional[float]) – dropout ratio

Methods

forward

Defines the computation performed at every call.

Attributes