DGL_图的创建、保存、加载

import dgl
import torch as th
from dgl.data.utils import save_graphsg1 = dgl.DGLGraph()
g1.add_nodes(3)
g1.add_edges([0, 0, 0, 1, 1, 2], [0, 1, 2, 1, 2, 2])
g1.ndata["x"] = th.ones(3, 5)   # 3个节点的embedding
g1.edata['y'] = th.zeros(6, 5)  # 6条边的embedding
# 补充：添加边的方式
# g1.add_edges(th.tensor([3, 4, 5]), 1)  # three edges: 3->1, 4->1, 5->1
# g1.add_edges(4, [7, 8, 9])  # three edges: 4->7, 4->8, 4->9
# g1.add_edges([1, 2, 3], [3, 4, 5])  # three edges: 1->3, 2->4, 3->5g2 = dgl.DGLGraph()
g2.add_nodes(3)
g2.add_edges([0, 1, 2], [1, 2, 1])
g2.edata["e"] = th.ones(3, 4)graph_labels = {"graph_sizes": th.tensor([3, 3])}save_graphs("data/try1.bin", [g1, g2], graph_labels)

from dgl.data.utils import load_graphs
from dgl.data.utils import load_labels# glist, label_dict = load_graphs("data/small.bin") # glist will be [g1, g2]
glist, label_dict = load_graphs("data/try1.bin", [0]) # glist will be [g1]
graph_sizes = load_labels("data/try1.bin")print(glist)
# [DGLGraph(num_nodes=3, num_edges=6,
#          ndata_schemes={'x': Scheme(shape=(5,), dtype=torch.float32)}
#          edata_schemes={'y': Scheme(shape=(5,), dtype=torch.float32)})]
print(label_dict)
# {'graph_sizes': tensor([3, 3])}
print(graph_sizes)
# {'graph_sizes': tensor([3, 3])}

DGL_图的创建、保存、加载相关推荐

Python图片转gif(将静态图转化为分块加载的动态图)
简介将静态图转化为分块加载的动态图方案 1. PIL: 1. 创建背景图2. 将原图拆分成N块并依次合成到背景图的相应位置, 得到N张素材图3. 将N张素材图合成GIF2. pygifsicle对 ...
5.1 Tensorflow:图与模型的加载与存储
这里写目录标题前言快速应用存储与加载,简单示例存储的文件 tf.train.Saver与存储文件的讲解核心定义存储文件的讲解保存图与模型进阶按迭代次数保存按时间保存更详细的解释前 ...
word2vec模型训练保存加载及简单使用
目录 word2vec模型训练保存加载及简单使用一 word2vec简介二.模型训练和保存及加载模型训练模型保存和加载模型的增量训练三.模型常用API 四.文本相似度计算--文档级别 wo ...
orb_slam3实现保存/加载地图功能and发布位姿功能
1.保存/加载地图先说方法:在加载的相机参数文件.yaml的最前面加上下面两行就行. System.LoadAtlasFromFile: "MH01_to_MH05_stereo_iner ...
unity保存加载慢_掌握Unity 5中的保存和加载功能
unity保存加载慢 Thanks to Vincent Quarles for kindly helping to peer review this article. 感谢Vincent Quarl ...
前端|npm全局创建wps加载项
欢迎点击「算法与编程之美」↑关注我们! 本文首发于微信公众号:"算法与编程之美",欢迎关注,及时了解更多此系列文章. 问题描述基于vscode的方式创建wps加载项在获取类型之后 ...
docker保存linux镜像,docker导入导出容器和保存加载镜像
系统环境:centos7.4 版本: # docker -v 1.docker容器导入导出 (1)查看:# docker ps -a (2)导出---export # docker export we ...
R语言保存加载工作空间或者工作空间数据对象实战（Save Load RData Workspace）
R语言保存加载工作空间或者工作空间数据对象实战(Save & Load RData Workspace) 目录 R语言保存加载工作空间或者工作空间数据对象实战(Save & Load ...
【Android 插件化】“ 插桩式 “ 插件化框架 ( 类加载器创建 | 资源加载 )
Android 插件化系列文章目录 [Android 插件化]插件化简介 ( 组件化与插件化 ) [Android 插件化]插件化原理 ( JVM 内存数据 | 类加载流程 ) [Android 插件 ...
Matlab停在载入界面,试图在Matlab用户界面中实现保存/加载对象功能时遇到了困难...
我尝试在Matlab(R2009A)用户界面中实现保存/加载函数.我的对象实现了一个布局函数,它为对象生成一个用户界面.我正在尝试实现保存/加载按钮的回调."保存"按钮起作用,并将 ...

DGL_图的创建、保存、加载

DGL_图的创建、保存、加载相关推荐

最新文章

热门文章