The configuration of my work as following:
The problem is that when I create a heterograph, I can feed dgl.heterograph() a data_dict, however, the data_dict is the information of the edge in the aspect of the node. Can I convey the weight feature of the edge in this data_dict? Or is there other more convienient ways to solve this need?
Part of the code is as follows:
import dgl import numpy as np import torch as th from dgl.nn import EdgeWeightNorm, GraphConv from collections import defaultdict a=[0,1,2,3,2,5] b=[1,2,3,4,0,3] edge_wights = [0.1,0.3,0.004,0.5,1,0.7] data_dict = defaultdict(list) for i in range(6): data_dict[('h_type', i, 'h_type')].append((a[i],b[i])) g=dgl.heterograph(data_dict) print(g)