-
Notifications
You must be signed in to change notification settings - Fork 9
/
README.Rmd
50 lines (37 loc) · 2.2 KB
/
README.Rmd
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
---
output: github_document
---
<!-- README.md is generated from README.Rmd. Please edit that file -->
```{r, include = FALSE}
knitr::opts_chunk$set(
collapse = TRUE,
comment = "#>",
fig.path = "man/figures/README-",
out.width = "100%"
)
```
# tft
<!-- badges: start -->
[![R build status](https://github.com/mlverse/tft/workflows/R-CMD-check/badge.svg)](https://github.com/mlverse/tft/actions)
[![Lifecycle: experimental](https://img.shields.io/badge/lifecycle-experimental-orange.svg)](https://www.tidyverse.org/lifecycle/#experimental)
[![CRAN status](https://www.r-pkg.org/badges/version/tft)](https://CRAN.R-project.org/package=tft)
[![](https://cranlogs.r-pkg.org/badges/tft)](https://cran.r-project.org/package=tft)
[![Codecov test coverage](https://codecov.io/gh/mlverse/tft/branch/master/graph/badge.svg)](https://codecov.io/gh/mlverse/tft?branch=master)
<!-- badges: end -->
> An R implementation of [tft: Temporal Fusion Transformer](https://arxiv.org/pdf/1912.09363.pdf).
The Temporal Fusion Transformer is a neural network architecture proposed by Bryan Lim et al. with the goal of making multi-horizon time series forecasts for
multiple time series in a single model.
The main difference between TFT and conventional forecasting methodologies is the
way its architecture allows encoding different types of input data that can exist
in forecasting problems. For instance, the model allows handling static covariates
and time varying (known and unknown) differently. tft also showed [promising benchmarks](https://ai.googleblog.com/2021/12/interpretable-deep-learning-for-time.html).
The code in this repository is heavily inspired in code from [akeskiner/Temporal_Fusion_Transform](https://github.com/akeskiner/Temporal_Fusion_Transform), [jdb78/pytorch-forecasting](https://github.com/jdb78/pytorch-forecasting) and
the original implementation [here](https://github.com/google-research/google-research/tree/master/tft).
## Installation
You can install the development version [GitHub](https://github.com/) with:
```r
# install.packages("remotes")
remotes::install_github("mlverse/tft")
```
Read the [Getting Started](https://mlverse.github.io/tft/articles/Getting-started.html) guide to fit your first
model with tft.