当前位置: 首页 > article >正文

torch_geometric使用手册-Creating Graph Datasets(专题四)

虽然 PyG 已经提供了很多有用的数据集,但你可能希望创建自己的数据集,使用自己收集的数据或非公开的数据。

自己实现数据集是非常简单的,你可能想查看源代码,了解各种数据集是如何实现的。然而,这里简要介绍一下如何设置你自己的数据集。

我们提供了两个抽象类用于数据集:torch_geometric.data.Datasettorch_geometric.data.InMemoryDatasettorch_geometric.data.InMemoryDataset 继承自 torch_geometric.data.Dataset,如果整个数据集能够加载到 CPU 内存中,就应该使用 InMemoryDataset

按照 torchvision 的约定,每个数据集都会传入一个根目录(root folder),该目录指定了数据集的存储位置。我们将根目录分成两个子目录:raw_dir,数据集会下载到此目录;processed_dir,处理后的数据集会保存在此目录。

此外,每个数据集还可以传入 transformpre_transformpre_filter 函数,默认情况下它们是 N


http://www.kler.cn/a/419598.html

相关文章:

  • 华为HarmonyOS 让应用快速拥有账号能力 -- 1 华为账号一键登录
  • yolov11剪枝
  • 【AI系统】代数简化
  • Linux-虚拟环境
  • Docker扩容操作(docker总是空间不足)
  • 【ONE·基础算法 || 动态规划(三)】
  • 《C++ Primer Plus》学习笔记|第1章 预备知识 (24-12-2更新)
  • Python的学习路线
  • opensatck-海光C86芯片的计算节点直通英伟达T4 GPU加速卡的操作记录
  • Vue单页面应用和多页面应用
  • Python3 多线程使用concurrent.futures
  • tomcat 8在idea启动控制台乱码
  • 四元数Quaternion的概念以及应用领域和处理方法
  • 无法通过windows功能控制面板自动安装或卸载windows server角色和功能
  • 【人工智能】用Python实现卷积神经网络(CNN)进行图像分类:从零开始的深度学习教程
  • Linux 各个目录作用
  • 如何从 Hugging Face 数据集中随机采样数据并保存为新的 Arrow 文件
  • 会展观众数据收集与分析
  • PGSQL:联合唯一索引的创建和删除
  • SpringBoot 整合 Avro 与 Kafka
  • Android——android相对布局(RelativeLayout)及各属性
  • go语言切片
  • 鸿蒙开发-ArkTs中实现websocket功能
  • 使用 Spring AOP 实现日志记录
  • Java 语言的起源发展与基本概念(JDK,JRE,JVM)
  • 基于Java Springboot电子书阅读器APP且微信小程序