【dgl学习】dgl.dataloader中链接预测任务的负采样方法neg_sampler详解

�小台

关注

于 2023-09-26 20:52:03 发布 · 890 阅读

原创

本文为博客 VIP 文章，开通 VIP 后可阅读全文

开通 VIP

dgl分批次进行链接预测任务时，会用到dataloader类，可以参考一下代码定义自己的dataloader。

dataloader = dgl.dataloading.DistEdgeDataLoader(
# dataloader = dgl.dataloading.EdgeDataLoader()
    graph, train_seeds, sampler,
    negative_sampler=dgl.dataloading.negative_sampler.Uniform(5),
    # negative_sampler=NegativeSampler(graph,args.negsample),<