1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +08:00
This commit is contained in:
Jactus
2020-11-28 10:50:39 +08:00
parent d820d5b8f6
commit eaa9f1a80d
2 changed files with 1 additions and 4 deletions

View File

@@ -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.

View File

@@ -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):
"""