assign train and validation masks for heterogeneous nodes to split my heterogeneous graph into train, validation and test sets.
this is my graph:
G = dgl.heterograph({
('user', 'rates', 'item') : M.nonzero(),
('item', 'rated_by', 'user') : M.transpose().nonzero()})
G.nodes['user'].data['x'] = users_features
G.nodes['item'].data['x'] = items_features
G.edges['rates'].data['h'] = weights
and thank you