Bering.models.BaselineMLP.get_latent

BaselineMLP.get_latent(data, num_mlp_layers_remain=2)[source]

Get latent features

Parameters:

data – Pytorch Geometric Data object
num_mlp_layers_remain – The number of layers from the last MLP layer. The output of this layer is the latent features.