self.y should bu related to negative_edges and test_negative_edges,
when we calculate "calculate_regression_loss" function,
pos = torch.cat((self.positive_z_i, self.positive_z_j), 1) # [14624, 128]
neg = torch.cat((self.negative_z_i, self.negative_z_j), 1) # [2522, 128]
thanks for your code! I have a little question,
self.y should bu related to negative_edges and test_negative_edges, when we calculate "calculate_regression_loss" function, pos = torch.cat((self.positive_z_i, self.positive_z_j), 1) # [14624, 128] neg = torch.cat((self.negative_z_i, self.negative_z_j), 1) # [2522, 128]