.. automodule:: stable_gnn.embedding.model_train_embeddings
   :members:
   :undoc-members:
   :show-inheritance: