From e25fc3b243728d7c4268df1ddc3c9ccdc500ed91 Mon Sep 17 00:00:00 2001 From: Alex Wang Date: Thu, 26 Nov 2020 22:28:34 +0800 Subject: [PATCH] update config --- qlib/contrib/model/pytorch_sfm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/pytorch_sfm.py b/qlib/contrib/model/pytorch_sfm.py index 7fbbd7c6e..e013238fa 100644 --- a/qlib/contrib/model/pytorch_sfm.py +++ b/qlib/contrib/model/pytorch_sfm.py @@ -340,7 +340,7 @@ class SFM(Model): def train_epoch(self, x_train, y_train): x_train_values = x_train.values - y_train_values = np.squeeze(y_train.values) * 100 + y_train_values = np.squeeze(y_train.values) self.sfm_model.train()