Fusionne dev dans test/integration-api-db-ml
La PR #123 (MLflow) est arrivée sur dev entre-temps. Un seul conflit, la liste .PHONY du Makefile : elle garde `migrate-test` d'ici et `mlflow-up` de dev, les deux cibles existant chacune de leur côté. Rien d'autre ne se recoupe : le test de chaîne passait déjà son propre `--mlflow-tracking-uri` sur un SQLite jetable, et `modele_jetable` entraîne son Booster sans passer par `train()`, qui journalise dans MLflow sans garde.
This commit is contained in:
@@ -181,7 +181,11 @@ def _log_to_mlflow(
|
||||
)
|
||||
mlflow.log_metrics({f"model_{cle}": valeur for cle, valeur in model_metrics.items()})
|
||||
mlflow.log_metrics({f"baseline_{cle}": valeur for cle, valeur in baseline_metrics.items()})
|
||||
mlflow.lightgbm.log_model(booster, name="model")
|
||||
mlflow.lightgbm.log_model(
|
||||
booster,
|
||||
name="model",
|
||||
registered_model_name="consumption-forecast-lightgbm",
|
||||
)
|
||||
mlflow.log_artifact(str(model_output))
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user