Update training setting.

2026-07-14 08:16:54 +08:00 · 2020-11-27 10:09:17 +08:00
parent 13c2c41d23
commit 6fb19eb58d
11 changed files with 91 additions and 50 deletions
--- a/qlib/contrib/model/pytorch_gru.py
+++ b/qlib/contrib/model/pytorch_gru.py
@@ -146,6 +146,7 @@ class GRU(Model):

        raise ValueError("unknown metric `%s`" % self.metric)

+
    def train_epoch(self, x_train, y_train):

        x_train_values = x_train.values