unit8co/darts

[Feature] Improve handling & documentation of encoders and categorical static covariates in .fit_from_dataset()

Open

#3006 aperta il 4 feb 2026

Vedi su GitHub
 (3 commenti) (1 reazione) (0 assegnatari)Python (762 fork)batch import
documentationgood first issue

Metriche repository

Star
 (6832 star)
Metriche merge PR
 (Merge medio 143g 16h) (16 PR mergiate in 30 g)

Descrizione

Describe the issue linked to the documentation I couldn't find a mention of this in docs but it appears that TorchForecastingModel doesn't use encoders when using fit/predict_from_dataset - it doesn't warn about it either during the call. This seems like a potentially big blind side for the user since it is a silent behavior. Since this method is pretty well documented otherwise, this seems like an unexpected caveat.

I wonder if this should be documented OR actually move encoder logic to be included in the _from_dataset method calls.

Sidenote: This leads to issues with TFTModel since it just skips training and errors on predict if datasets don't contain future_covariates.

Sidenote2: categorical feature embeddings with TFTModel are also not working as expected (essentially ignored AFAIK) when using the _from_dataset methods since the code uses feature integer index instead of names for lookups but this seems like a completely separate issue.

Guida contributor