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

Update pytorch_sfm.py

Browse Source
This commit is contained in:
Haoyu Wang
2020-11-27 09:59:30 +08:00
committed by GitHub
parent d5adc4934b
commit 13c2c41d23
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -102,7 +102,7 @@ class SFM_Model(nn.Module):
i = self.inner_activation( i = self.inner_activation(
x_i + torch.matmul(h_tm1 * B_U[0], self.U_i) x_i + torch.matmul(h_tm1 * B_U[0], self.U_i)
) # not sure whether I am doing in the right unsquuze )
ste = self.inner_activation(x_ste + torch.matmul(h_tm1 * B_U[0], self.U_ste)) ste = self.inner_activation(x_ste + torch.matmul(h_tm1 * B_U[0], self.U_ste))
fre = self.inner_activation(x_fre + torch.matmul(h_tm1 * B_U[0], self.U_fre)) fre = self.inner_activation(x_fre + torch.matmul(h_tm1 * B_U[0], self.U_fre))