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

目录

  • 一、各部分code以及输出
    • 1.1 数据集 下载与查看
    • 1.2 试试应用 传统全连接层
    • 1.3 每轮的训练,损失值输出
    • 1.4 准确率计算
    • 1.5 将全连接层 替换为 GCN层
    • 1.6 训练GCN模型,准确率计算
  • 二、完整code 整合

一、各部分code以及输出

  • 所用到的数据集:Cora dataset

  • 数据集的描述官网:数据集描述:Yang et al. (2016)

最开始运行时,可能 会遇到 部分 因为网络不稳定(或许是)而出现的错误:

【问题记录与解决】TimeoutError: [WinError 10060] 由于连接方在一段时间后没有正确答复或连接的主机没有反应,连接尝试失败。

1.1 数据集 下载与查看

来看code以

你可能感兴趣的:(【小小的项目,(实战+案例)】,transformer,点分类任务,GCN,图神经网络,人工智能)