首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

【知识】DGL中graph默认的稀疏矩阵格式和coo格式不对的坑

fraud数据集,是先从文件读取矩阵,然后转为了coo: 因此,得出结论:对于dgl的数据集,则取决于数据集的npz文件中指定的格式,或数据集自己的处理方式;代码验证 dgl.DGLGraph.formats...— DGL 0.8.2post1 documentation 对于formats这个函数: 如果 formats 为 None,则返回稀疏格式的使用状态;否则,可以是'coo'/'csr'/'csc'...自己用graph接口的方式:import dglg = dgl.graph(([0, 1, 2], [1, 2, 3]))print(g.formats())# 输出:{'created': ['coo...created': ['csr', 'csc']}g = dgl.graph(('csr', ([0, 0, 0, 1, 2, 3], [1, 2, 3], [0, 1, 2])))print(g.formats...再看一下Reddit,确实又是coo格式的:import dgldataset = dgl.data.RedditDataset()g = dataset[0]print(g.formats())# 输出

5010
领券