Thanks for the code! I'm looking for the embeddings/representations for each node in python code. Please help me through.