GeneralEncoder

class marius.nn.encoders.GeneralEncoder

__init__(self: marius._nn.encoders.GeneralEncoder, encoder_config: marius._config.EncoderConfig, device: torch.device, num_relations: int = 1) → None

__init__(self: marius._nn.encoders.GeneralEncoder, layers: List[List[Layer]]) → None

forward(self: marius._nn.encoders.GeneralEncoder, embeddings: Optional[torch.Tensor], features: Optional[torch.Tensor], dense_graph: marius._data.DENSEGraph, train: bool = True) → torch.Tensor

property device

property encoder_config

property layers

property num_relations

reset(self: marius._nn.encoders.GeneralEncoder) → None