From 5c1a8184883460bb322171cb909606354264ede8 Mon Sep 17 00:00:00 2001 From: Wendi Li Date: Wed, 25 Nov 2020 20:09:47 +0800 Subject: [PATCH] Update README.md --- examples/benchmarks/TFT/README.md | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/benchmarks/TFT/README.md b/examples/benchmarks/TFT/README.md index 03d45e087..6d605a1bd 100644 --- a/examples/benchmarks/TFT/README.md +++ b/examples/benchmarks/TFT/README.md @@ -1,10 +1,12 @@ # Temporal Fusion Transformers Benchmark ## Source -https://github.com/google-research/google-research/tree/master/tft +**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. The model must run in GPU, or an error will be raised. -2. New datasets should be registered in ``data_formatters``, for detail please visit the source. \ No newline at end of file +2. New datasets should be registered in ``data_formatters``, for detail please visit the source.