(self, inputs)
| 63 | raise NotImplementedError |
| 64 | |
| 65 | def to_sample(self, inputs): |
| 66 | batch_size = tf.size(inputs) |
| 67 | src = tf.expand_dims(inputs, -1) |
| 68 | pos = tf_euler.sample_neighbor(inputs, self.edge_type, 1, |
| 69 | self.max_id + 1)[0] |
| 70 | negs = tf_euler.sample_node(batch_size * self.num_negs, self.node_type) |
| 71 | negs = tf.reshape(negs, [batch_size, self.num_negs]) |
| 72 | return src, pos, negs |
| 73 | |
| 74 | def __call__(self, inputs): |
| 75 | src, pos, negs = self.to_sample(inputs) |