def collate(self, batch):
packed = self.pack(batch)
y = packed.pop("y_list", None)
if y is not None:
y = collate_labels_disjoint(y, node_level=self.node_level)
output = to_disjoint(**packed)
output = sp_matrices_to_sp_tensors(output)
if len(output) == 1:
output = output[0]
if y is None:
return output
else:
return output, y
def collate(self, batch):
packed = self.pack(batch)
y = packed.pop("y_list", None)
if y is not None:
y = collate_labels_disjoint(y, node_level=True)
output = to_disjoint(**packed)
output = output[:-1] # Discard batch index
output = sp_matrices_to_sp_tensors(output)
if len(output) == 1:
output = output[0]
output = (output, )
if y is not None:
output += (y, )
if self.sample_weights is not None:
output += (self.sample_weights, )
if len(output) == 1:
output = output[0] # Again, in case there are no targets and no SW
return output