liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-15 08:46:56 +08:00
Code Activity

change code style for gats

Browse Source
This commit is contained in:
Hong Zhang
2020-11-26 13:58:45 +08:00
parent 85f9d5594e
commit e8bb5061b0
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_gats.py
View File

@@ -370,7 +370,6 @@ class GATModel(nn.Module):
self.fc_out = nn.Linear(hidden_size, 1) self.fc_out = nn.Linear(hidden_size, 1)
self.leaky_relu = nn.LeakyReLU() self.leaky_relu = nn.LeakyReLU()
self.softmax = nn.Softmax(dim=1) self.softmax = nn.Softmax(dim=1)
self.d_feat = d_feat self.d_feat = d_feat
def cal_convariance(self, x, y): # the 2nd dimension of x and y are the same 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) out, _ = self.rnn(x)
hidden = out[:, -1, :] hidden = out[:, -1, :]
hidden = self.bn1(hidden) hidden = self.bn1(hidden)
gamma = self.cal_convariance(hidden, hidden) gamma = self.cal_convariance(hidden, hidden)
output = gamma.mm(hidden) output = gamma.mm(hidden)
output = self.fc(output) output = self.fc(output)