1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-13 07:46:53 +08:00

Add random seed.

This commit is contained in:
lwwang1995
2020-11-30 20:51:19 +08:00
committed by you-n-g
parent d6ff62b48c
commit e0dad0e85a
5 changed files with 19 additions and 0 deletions

View File

@@ -113,6 +113,9 @@ class GRU(Model):
)
)
np.random.seed(self.seed)
torch.manual_seed(self.seed)
self.gru_model = GRUModel(
d_feat=self.d_feat,
hidden_size=self.hidden_size,