rl_equation_solver.utilities.utilities.GraphEmbedding

class GraphEmbedding(graph, n_observations, n_features, device)

Bases: object

Graph embedding class for embedding node features in matrix of fixed sizes

Methods