Skip to main content

Training with MLflow

Train and register a model while preserving data, parameters, metrics, and artifacts.

Training with MLflow

Activity responsibilities

ActivityResponsibility
ingest-dataExecute the ingest data stage and publish its declared outputs for downstream activities.
validateExecute the validate stage and publish its declared outputs for downstream activities.
trainExecute the train stage and publish its declared outputs for downstream activities.
evaluateExecute the evaluate stage and publish its declared outputs for downstream activities.
register-modelExecute the register model stage and publish its declared outputs for downstream activities.
log-mlflow-runExecute the log MLflow run stage and publish its declared outputs for downstream activities.

Inputs

  • Versioned dataset
  • Training configuration
  • Runtime image

Outputs

  • Model artifact
  • Evaluation report
  • MLflow run

Provenance to preserve

  • Dataset and code digests
  • Parameters and metrics
  • Resource placement

Runnable example

This is the complete checked-in bundle for this pattern. Download the environment, scope, topology, workflow, input, container recipe, runner, and validation contract from this page before executing it.

Verified local execution

These captures and the output manifest were produced by the fixture's local Docker run and validator. They are published with the same bundle as the runnable files.

Training with MLflow workflow execution evidence
Workflow evidence
Training with MLflow execution evidence
Execution evidence
Training with MLflow output evidence
Output evidence
Open verified output manifest ↓

Execution considerations

Use GPU capacity for training and CPU capacity for validation or registration. MLflow remains an external service; AkôFlow records the activity that communicates with it.

AkôFlow boundary: the engine schedules, deploys, executes, transfers data, and records evidence. The ML or agent framework remains an implementation choice inside each activity.