网站首页 站内搜索

搜索结果

查询Tags标签: InMemoryDataset,共有 1条记录
  • 图神经网络之四:InMemoryDataset基类与节点预测与边预测

    图神经网络之四:InMemoryDataset基类与节点预测与边预测引言 这篇是datawhale组队学习之图神经网络第四篇,本笔记主要梳理课程的关键点,以及简单的代码实现。 InMemoryDataset基类 在PyG中,我们通过继承InMemoryDataset类来自定义一个数据可全部存储到内存的数据集类。…

    2021/6/28 6:21:31 人评论 次浏览
扫一扫关注最新编程教程