Skip to content

Make forecast boosting rounds configurable - #14

Merged
amarakramali merged 1 commit into
mainfrom
agent/configurable-forecast-estimators
Jul 27, 2026
Merged

Make forecast boosting rounds configurable#14
amarakramali merged 1 commit into
mainfrom
agent/configurable-forecast-estimators

Conversation

@amarakramali

Copy link
Copy Markdown
Owner

What changed

  • add --estimators to the production forecast CLI
  • forward the selected count to final XGBoost training
  • document the option in the configurable forecast example
  • add a command-level regression test covering option forwarding and outputs

Why

The evaluator and demo already allowed the number of boosting rounds to be
configured, and the README said this setting was configurable. The production
aep-forecast command nevertheless always used the hard-coded default of 800.

User impact

Users can now trade training time against model capacity consistently across
evaluation, demo, and production forecast commands.

Validation

  • python -m pytest -q (66 passed)
  • python -m ruff check src tests streamlit_app.py
  • python -m compileall -q src tests streamlit_app.py
  • python -m build
  • verified installed aep-forecast --help exposes --estimators

@amarakramali
amarakramali merged commit a53cea8 into main Jul 27, 2026
2 checks passed
@amarakramali
amarakramali deleted the agent/configurable-forecast-estimators branch July 27, 2026 15:56
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.

1 participant