diff --git a/README.md b/README.md index 884bbb5c0..123134319 100644 --- 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 index a3b4fc919..3387a5947 100644 --- a/examples/benchmarks/TFT/tft.py +++ b/examples/benchmarks/TFT/tft.py @@ -233,9 +233,8 @@ class TFTModel(ModelFT): 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): """