Unverified Commit 76bb5404 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Black auto fix. (#4682)


Co-authored-by: default avatarSteve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
parent a208e886
This diff is collapsed.
...@@ -3,6 +3,7 @@ ...@@ -3,6 +3,7 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
class TransE(nn.Module): class TransE(nn.Module):
r"""Similarity measure from `Translating Embeddings for Modeling Multi-relational Data r"""Similarity measure from `Translating Embeddings for Modeling Multi-relational Data
<https://papers.nips.cc/paper/2013/hash/1cecc7a77928ca8133fa24680a88d2f9-Abstract.html>`__ <https://papers.nips.cc/paper/2013/hash/1cecc7a77928ca8133fa24680a88d2f9-Abstract.html>`__
...@@ -53,6 +54,7 @@ class TransE(nn.Module): ...@@ -53,6 +54,7 @@ class TransE(nn.Module):
>>> scorer(h_head, h_tail, rels).shape >>> scorer(h_head, h_tail, rels).shape
torch.Size([30]) torch.Size([30])
""" """
def __init__(self, num_rels, feats, p=1): def __init__(self, num_rels, feats, p=1):
super(TransE, self).__init__() super(TransE, self).__init__()
...@@ -94,4 +96,4 @@ class TransE(nn.Module): ...@@ -94,4 +96,4 @@ class TransE(nn.Module):
""" """
h_rel = self.rel_emb(rels) h_rel = self.rel_emb(rels)
return - torch.norm(h_head + h_rel - h_tail, p=self.p, dim=-1) return -torch.norm(h_head + h_rel - h_tail, p=self.p, dim=-1)
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment