benchopt / benchopt/benchmark_tsfm

Implement `embed` and `time_embed` for the `BaseTSFMSolver`

Open
#39 1 comment 1 reaction 0 assignees View on GitHub
Dominant language
Python
Stars
15
Forks
21
PR merge metrics
No merged PRs in 30d

Description

A first attempt at simplifying the code for models has been to code a new parent `BaseTSFMSolver` class in which most of the logics for forecasting is located (such that each new model would just need to inherit + implement `batch_forecast` and a model loader).

The next step would be to do the same for `embed` and `time_embed`, see discussion #31
That means having `batch_embed` and `batch_time_embed` (signature defined in `BaseTSFMSolver` with tensor shapes specified in the docs, actual implementation deferred per model).

Once this is done, a full pass on the whole set of available models should be done to make sure they stick (if it makes sense at least) to that behaviour.
Also, Chronos vs Chronos-v2 is a mess at the time, this will need to be improved at some point.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start with the BaseTSFMSolver definition and discussion #31 to confirm the batch_embed and batch_time_embed signatures and documented tensor shapes. Review the available model implementations, add the deferred per-model behavior, and check that the models consistently follow the new embedding interface where applicable.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Refactor
Difficulty
4/5
Estimated time
3-5 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
48/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.