Fix tft

lyrl · Nov 28, 2020 · eaa9f1a · eaa9f1a
1 parent d820d5b
commit eaa9f1a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/README.md b/README.md
@@ -201,9 +201,7 @@ Here is a list of models built on `Qlib`.
 - [LSTM based on pytorcn](qlib/contrib/model/pytorch_lstm.py)
 - [ALSTM based on pytorcn](qlib/contrib/model/pytorch_alstm.py)
 - [GATs based on pytorch](qlib/contrib/model/pytorch_gats.py)
-- [TabNet based on pytorch](qlib/contrib/model/tabnet.py)
 - [SFM based on pytorch](qlib/contrib/model/pytorch_sfm.py)
-- [HATs based on pytorch](qlib/contrib/model/pytorch_hats.py)
 - [TFT based on tensorflow](examples/benchmarks/TFT/tft.py)
 
 Your PR of new Quant models is highly welcomed.

diff --git a/examples/benchmarks/TFT/tft.py b/examples/benchmarks/TFT/tft.py
@@ -233,9 +233,8 @@ def predict(self, dataset):
             tf.keras.backend.set_session(default_keras_session)
 
         predict = format_score(p90_forecast, "pred", 0)  # self.label_shift
-        label = format_score(targets, "label", 0)
         # ===========================Predicting Process===========================
-        return predict, label
+        return predict
 
     def finetune(self, dataset: DatasetH):
         """