【Transformer 练习】图分类任务(单步+整合code)

声明:仅学习使用~

建议使用 Jupyter Notebook。当然PyCharm也可。

目录

  • 一、各部分code以及输出
    • 1.1 数据集概述
    • 1.2 Mini-batching
    • 1.3 设计网络结构,训练。
  • 二、完整code

一、各部分code以及输出

首先是数据集的下载方式:数据集下载所在链接(感兴趣的 看看就好,不需要主动下载,因为下面程序运行时,会主动下载的)

1.1 数据集概述

  • 这回我们拿到的不是一个图,而是一堆图(188个)
  • 对这些图整体进行分类任务,数据集概述如下:
# 昵 称:XieXu
# 时 间: 2022/12/25/0025 8:46
import torch
from torch_geometric

你可能感兴趣的:(transformer,图分类任务,图神经网络,人工智能,torch)