cosmogato
cosmogato

Reputation: 21

Pytorch Forecasting vs Darts, experiences welcome

I was wondering which package to use between pytorch forecasting (https://pytorch-forecasting.readthedocs.io/en/stable/) or darts (https://unit8co.github.io/darts/). I have been trying both, it looks like darts is more sklearn-like in its writing and style and pytorch forescasting uses different data classes.

Any comment comparing the two would be welcome.

I don't know if some of you might have carried out a performance comparison between both libraries.

Thanks in advance!

Upvotes: 2

Views: 1879

Answers (2)

alholz
alholz

Reputation: 11

If helpful, I have just made a pipeline for a Temporal Fusion Transformer model in Darts and am considering rebuilding it in PytorchForecasting for the following reasons:

  • Darts doesn't yet support output of variable importance from the TFT model (at least I haven't been able to figure it out)
  • Better support for static categorical features

As mentioned above, the dataset handling in Darts is pretty good and they have abstracted away the Pytorch dataloader

Upvotes: 1

chandler_liu
chandler_liu

Reputation: 19

I think one of the biggest advantage of darts is its Timeseries Object which is very pandas-like and very intuitive when you are familiar with sklearn. However, I also do see the advantage that pytorch-forecasting dealt with categorical data "better" (easier) and it takes a steeper learning curve to understand pytorch-forecasting. I would say pytorch-forecasting sometimes outperform darts using the same model.

Upvotes: 1

Related Questions