From e8bb5061b06da31942bd432c991ea6c5d431d434 Mon Sep 17 00:00:00 2001 From: Hong Zhang Date: Thu, 26 Nov 2020 13:58:45 +0800 Subject: [PATCH] change code style for gats --- qlib/contrib/model/pytorch_gats.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/qlib/contrib/model/pytorch_gats.py b/qlib/contrib/model/pytorch_gats.py index aa5b22119..7cdfb571a 100755 --- a/qlib/contrib/model/pytorch_gats.py +++ b/qlib/contrib/model/pytorch_gats.py @@ -370,7 +370,6 @@ class GATModel(nn.Module): self.fc_out = nn.Linear(hidden_size, 1) self.leaky_relu = nn.LeakyReLU() self.softmax = nn.Softmax(dim=1) - self.d_feat = d_feat def cal_convariance(self, x, y): # the 2nd dimension of x and y are the same @@ -389,7 +388,6 @@ class GATModel(nn.Module): out, _ = self.rnn(x) hidden = out[:, -1, :] hidden = self.bn1(hidden) - gamma = self.cal_convariance(hidden, hidden) output = gamma.mm(hidden) output = self.fc(output)