relationPrediction
relationPrediction copied to clipboard
deepakn97
ACL 2019: Learning Attention-based Embeddings for Relation Prediction in Knowledge Graphs
Can you explain to me the meaning of this auxiliary edge (preferably with an example), I want to apply it, but I still can't understand its specific meaning.
` mask_indices = torch.unique(batch_inputs[:, 2]).cuda() mask = torch.zeros(self.entity_embeddings.shape[0]).cuda() mask[mask_indices] = 1.0 entities_upgraded = self.entity_embeddings.mm(self.W_entities) out_entity_1 = entities_upgraded + \ mask.unsqueeze(-1).expand_as(out_entity_1) * out_entity_1 `
Hi, can you open the code about how to draw the heatmap about the attention matrix? Thank you!
I think SpecialSpmmFunctionFinal 's forward section is intend to compute the row sum of sparse matrix ,and backward return the gradient for the sparse matrix's values,but I find torch.sparse might...
Hi, I just wonder why your model have to go throw layer SpecialSpmmFunctionFinal and what is the intent of this layer The layer forward is : ```python class SpecialSpmmFunctionFinal(torch.autograd.Function): """Special...
The project has not supported DataParallel and multi GPUs yet. I tried to add the codes myself but found several bugs while running the new version. 1. Line 106, layers.py:...
I have the following problems: RuntimeError: Unable to find a valid cuDNN algorithm to run convolution I changed the batch_size from 128 to 32 or less, but it still doesn't...
Hello, authors. I have learnt a lot from your code, while I have a question about the `bfs` function described as follows: I know the code of create_batch.py lines`#259-276` is...
Hi, thanks for your amazing work. I'm trying to retrieve node embedding from GAT and CONV checkpoints. What the right way to do that? I'm trying something like this: ```python...
