Skip to content

feat(model): HistGradientBoosting con split temporal y umbral F1 - #18

Merged
Sebithaz-dev merged 3 commits into
mainfrom
feat/hist-gradient-boosting
Oct 27, 2026
Merged

Sebithaz-dev merged 3 commits into
mainfrom
feat/hist-gradient-boosting

Conversation

@Sebithaz-dev

Copy link
Copy Markdown
Owner

Cambios

  • Reemplaza LogisticRegression por HistGradientBoostingClassifier.
  • Split temporal 70/10/20 por starttime (con opción random para comparar).
  • Nuevas features: hour/month/dayofweek cíclicas, is_rush_hour e is_night.
  • Optimización del umbral en validación maximizando F1; se persiste en transformers.pkl y metricas.json.
  • Encoders, station_freq y scaler se ajustan solo con train para evitar leakage.
  • Agrega pr_auc y validation_size a las métricas.
  • Pipeline y tests de features/split/umbral.

Comparación en el mismo test temporal (94,686 filas)

Modelo ROC AUC PR AUC F1 (largo) Recall Precision
LogisticRegression baseline 0.6179 0.4030 0.4458 0.6725 0.3334
HistGradientBoosting 0.6505 0.4455 0.4650 0.7035 0.3473

Umbral F1: 0.46. Gate superado: mejora PR-AUC y F1 sobre el baseline.

Validación

  • ruff check ml tests scripts
  • pytest -q → 42 passed
  • python -m ml.health --json --skip-s3 → ok

@sonarqubecloud

Copy link
Copy Markdown

@Sebithaz-dev
Sebithaz-dev merged commit 6fddad4 into main Oct 27, 2026
7 checks passed
@Sebithaz-dev
Sebithaz-dev deleted the feat/hist-gradient-boosting branch October 27, 2026 21:47
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