PyG|实现构建自己的数据集

最新推荐文章于 2024-06-30 16:19:48 发布

原创最新推荐文章于 2024-06-30 16:19:48 发布 · 585 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #机器学习 #人工智能

深度学习同时被 2 个专栏收录

11 篇文章

订阅专栏

图神经网络

7 篇文章

订阅专栏

博客主要讲述了在torch中构建自己的数据集，涉及深度学习相关内容，为利用torch进行数据处理和模型训练提供基础。

在torch中构建自己的数据集。



import torch
from torch_geometric.data import Data

x = torch.tensor([[2,1],[5,6],[3,7],[12,0]], dtype=torch.float)
y = torch.tensor([0,1,0,1], dtype=float)

edge_index = torch.tensor([[0,1,2,0,3], # 起始点
                           [1,0,1,3,2]], dtype=torch.long) # 终止点
# 边的顺序打乱都可以


data = Data(x=x, y=y, edge_index=edge_index)

print(data)