mlverse / tft

R implementation of Temporal Fusion Transformers
https://mlverse.github.io/tft/
Other
26 stars 9 forks source link

Using predict with tft_fit model #14

Closed Boeinco closed 2 years ago

Boeinco commented 2 years ago

Hi I'm getting a single pred value repeated in list for my yhat predict using the example from the home page using vic_elec. This doesn't seem correct.

image

cregouby commented 2 years ago

you're absolutely right. That part is not finished yet.

Boeinco commented 2 years ago

Alright, thanks, no problem!

cregouby commented 2 years ago

@Boeinco , actually, the branch https://github.com/mlverse/tft/tree/feature/add_recipe_assertions should give you satisfaction (I'd forgot that) You may install it with remotes::install_github("mlverse/tft@feature/add_recipe_assertions")