Skip to content

feat: add support for more ml models#337

Merged
AzulGarza merged 14 commits intomainfrom
feat/ml-models
Apr 25, 2026
Merged

feat: add support for more ml models#337
AzulGarza merged 14 commits intomainfrom
feat/ml-models

Conversation

@AzulGarza
Copy link
Copy Markdown
Member

this pr adds support for more ml models

Copilot AI review requested due to automatic review settings April 25, 2026 20:00
Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Adds additional mlforecast-backed “Auto*” forecasters and expands probabilistic (quantile) forecasting support across the ML and NeuralForecast model wrappers.

Changes:

  • Added 7 new AutoMLForecast model wrappers (LinearRegression, XGBoost, Ridge, Lasso, ElasticNet, RandomForest, CatBoost) and refactored shared training/prediction into run_automlforecast_model.
  • Enabled quantiles support for AutoNHITS and AutoTFT using NeuralForecast’s MQLoss, plus standardized forecast output conversion via QuantileConverter.
  • Updated public exports, tests, dependencies, and changelogs to cover the new models and quantile behavior.

Reviewed changes

Copilot reviewed 8 out of 9 changed files in this pull request and generated 2 comments.

Show a summary per file
File Description
timecopilot/models/neural.py Adds quantile forecasting support for AutoNHITS/AutoTFT via MQLoss and output conversion.
timecopilot/models/ml.py Introduces multiple new ML wrappers and shared run_automlforecast_model helper with quantile output support.
timecopilot/models/__init__.py Exposes new ML forecasters via imports and __all__.
tests/models/test_models.py Expands tests to include new models and adjusts quantile/level expectations.
tests/models/conftest.py Registers new ML models in the test model matrix.
pyproject.toml Adds catboost and xgboost dependencies required by the new model wrappers.
docs/changelogs/v0.0.26.md Documents quantile support for AutoLGBM/AutoNHITS/AutoTFT.
docs/changelogs/v0.0.25.md Documents newly added ML models.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Comment thread tests/models/test_models.py Outdated
Comment thread docs/changelogs/v0.0.26.md Outdated
@AzulGarza AzulGarza merged commit 7193d1c into main Apr 25, 2026
10 checks passed
@AzulGarza AzulGarza deleted the feat/ml-models branch April 25, 2026 21:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants