def __init__(self,
x=None,
edge_index=None,
edge_attr=None,
y=None,
pos=None,
norm=None,
face=None,
**kwargs):
Data.__init__(self,
x=x,
edge_index=edge_index,
edge_attr=edge_attr,
y=y,
pos=pos,
norm=norm,
face=face,
**kwargs)
def __init__(self, x, edge_index, task_list):
Data.__init__(self, x, edge_index)
self.task_list = np.array(task_list)
self.task_to_num = {v: k for (k, v) in enumerate(self.task_list)}
def __init__(self, x, edge_index, task_list):
Data.__init__(self, x, edge_index.to(torch.long))
self.task_list = np.array(task_list)
self.task_to_num = {v: k for (k, v) in enumerate(self.task_list)}
self.n = len(self.x)
