sketchgraphs_models.graph.model.message_passing

This module contains the components of the graph model associated with handling the message passing.

Classes

DenseOnlyEmbedding(cardinality, dimension)	Generic pre-embedding module which encapsulates a pytorch embedding layer.
DenseSparsePreEmbedding(target_type, …[, …])	This is a generic pre-embedding module which combines dense and sparse pre-embeddings.
GraphModelCore(message_passing, …)	Component of the entity model used to compute global features, i.e. graph and node embeddings.
GraphPostEmbedding(hidden_size[, …])	Component of the graph model which computes graph-wide representation by aggregating node representations.