generate_mask_tensor

[ch3njust1n]

📚 Documentation

5.1 Node Classification/Regression
Please add the code for the generate_mask_tensor() function that's referenced. I can't find this in any of the tutorials. This would be very valuable to have. Thanks!

[HuXiangkun]

Hi, function generate_mask_tensor converts a numpy mask into a tensor of different backend for builtin datasets:

dgl/python/dgl/data/utils.py

Line 301 in edb9787

def generate_mask_tensor(mask):

So if user wants to create a dataset, the training, validation and testing mask are generated by the user.