oracle / oci-mlflow

The Oracle Cloud Infrastructure (OCI) MLflow plugin empowers users of OCI by providing seamless integration with OCI resources, allowing them to effectively manage the entire life cycle of their machine learning use cases.

OCI Mlflow Plugin

The OCI MLflow plugin enables OCI users to use OCI resources to manage their machine learning usecase life cycle. This table below provides the mapping between the MLflow features and the OCI resources that are used.

MLflow Use Case	OCI Resource
User running machine learning experiments on notebook, logs model artifacts, model performance etc	Data Science Jobs, Object Storage, MySQL
Batch workloads using spark	Data Flow, Object Storage, MySQL
Model Deployment	Data Science Model Deployment
User running machine learning experiments on notebook, logs model artifacts, model performance etc	Object Storage, MySQL

Installation

To install the oci-mlflow plugin call -

  python3 -m pip install oci-mlflow

To test the oci-mlflow plugin call -

  mlflow deployments help -t oci-datascience

Documentation

• OCI MLflow Documentation
• Getting started with Oracle Accelerated Data Science SDK
• Getting started with OCI Data Science Jobs
• Getting started with Data Science Environments
• Getting started with Custom Conda Environments
• Oracle AI & Data Science Blog
• OCI Documentation

Examples

Running MLflow projects on the OCI Data Science jobs and Data Flow applications -

export MLFLOW_TRACKING_URI=<tracking server url>
mlflow run . --experiment-name My-Experiment --backend oci-datascience --backend-config ./oci-datascience-config.json

Deploying MLflow models to the OCI Model Deployments -

mlflow deployments help -t oci-datascience

export MLFLOW_TRACKING_URI=<tracking server url>

mlflow deployments create --name <model deployment name> -m models:/<registered model name>/<model version> -t oci-datascience --config deploy-config-file=deployment_specification.yaml

Contributing

This project welcomes contributions from the community. Before submitting a pull request, pleasereview our contribution guide

Find Getting Started instructions for developers in README-development.md

Security

Consult the security guide SECURITY.md for our responsible security vulnerability disclosure process.

License

Copyright (c) 2023 Oracle and/or its affiliates. Licensed under the Universal Permissive License v1.0