def get_embeddings(model, data, device):
    """
    Get best embeddings
    
    Args
        - model (torch object): best model
        - data (Data object): dataset
        - device (torch object): cpu or cuda
    
    Return
        - all_emb (tensor): best embedding for all nodes 
    """
    model.eval() 
    data = data.to(device) 
    all_emb = model(data.x, data.edge_index) 
    print(all_emb.shape)
    return all_emb