liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +08:00
Code Activity
Files
2c21b8089a2f5d86893403445417992af2239168
qlib/examples/benchmarks/TFT
History
Linlang 3097dcc995 fix(security): use RestrictedUnpickler in load_instance (#2153) 
* fix(security): enforce RestrictedUnpickler for load_instance to prevent unsafe pickle deserialization

* fix: lint error
2026-03-10 20:45:38 +08:00
..
data_formatters
Adjust rolling api (#1594)
2023-07-14 12:16:12 +08:00
expt_settings
Adjust rolling api (#1594)
2023-07-14 12:16:12 +08:00
libs
fixed a problem with multi index caused by the default value of groupkey (#1917)
2025-05-13 16:02:49 +08:00
README.md
Merge nested main (#597)
2021-10-01 02:15:30 +08:00
requirements.txt
Merge nested main (#597)
2021-10-01 02:15:30 +08:00
tft.py
fix(security): use RestrictedUnpickler in load_instance (#2153)
2026-03-10 20:45:38 +08:00
workflow_config_tft_Alpha158.yaml
Adjust rolling api (#1594)
2023-07-14 12:16:12 +08:00

README.md

Temporal Fusion Transformers Benchmark

Source

Reference: Lim, Bryan, et al. "Temporal fusion transformers for interpretable multi-horizon time series forecasting." arXiv preprint arXiv:1912.09363 (2019).

GitHub: https://github.com/google-research/google-research/tree/master/tft

Run the Workflow

Users can follow the workflow_by_code_tft.py to run the benchmark.

Notes

  1. Please be aware that this script can only support Python 3.6 - 3.7.
  2. If the CUDA version on your machine is not 10.0, please remember to run the following commands conda install anaconda cudatoolkit=10.0 and conda install cudnn on your machine.
  3. The model must run in GPU, or an error will be raised.
  4. New datasets should be registered in data_formatters, for detail please visit the source.