Skip to content

aai-institute/ai-kickstart-mlops

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

48 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AI Career Kickstart MLOps example

Quickstart

The example requires Python 3.9 or later installed. If you want to host the MLOps tool stack (MLflow & lakeFS) on your local machine, Docker (and Docker Compose) is required.

Follow these steps in a local clone of the repository to reproduce the example Dagster workflow:

  • Create a Python virtual environment and install dependencies (macOS/Linux): python -m venv .venv && source .venv/bin/activate && pip install -r requirements.txt -e .
  • Optionally (not required when running locally with Docker Compose): Set MLFlow tracking URL and credentials and lakeFS repository information in src/ames_housing/constants.py
  • Start MLOps tool stack in Docker (in a separate shell): docker compose -f stack/docker-compose.yml up
  • Run Dagster: dagster dev -m ames_housing
  • Access Dagster web UI: http://localhost:3000
  • Click the Materialize all button to run the model training pipeline
  • Observe the tracked experiment in MLflow: http://localhost:5000 (if using Docker Compose)

Using the lakeFS I/O Managers

By default, data and models created by the workflow are persisted in the file system (under the data/ and model/ directories). If you want to persist these assets in lakeFS instead, set the ENV environment variable to production when running Dagster:

ENV=production dagster dev -m ames_housing

Make sure to create a repository named ai-kickstart in the lakeFS web UI before materializing the assets in Dagster (the Docker Compose setup uses the default lakeFS quickstart credentials for login: Access key ID AKIAIOSFOLQUICKSTART, secret access key wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY).

Then, add your lakeFS credentials to the .lakectl.yaml config file in your home directory to allow the Dagster I/O managers to automatically discover the lakeFS server URL and credentials (this step is only necessary once):

credentials:
  access_key_id: AKIAIOSFOLQUICKSTART
  secret_access_key: wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY
server:
  endpoint_url: http://127.0.0.1:8000

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •  

Languages