Skip to content

Backend implementation for running MLFlow projects on Hadoop/YARN.

License

Notifications You must be signed in to change notification settings

kiminh/mlflow-yarn

 
 

Repository files navigation

mlflow-yarn

Backend implementation for running MLFlow projects against a Hadoop/YARN backend).

To activate just do 'pip install mlflow-yarn' along your mlflow environment. It will register the plugin as an entrypoint with the YARN backend.

$ pip install mlflow-yarn

mlflow-yarn only supports Python ≥3.6.

Developement

Install from source

$ git clone https://github.com/criteo/cluster-pack
$ cd cluster-pack
$ pip install .

Examples

Example with pip only project

  • Dependencies are pulled from requirements.txt
$ git clone https://github.com/criteo/mlflow-yarn
$ pip install mlflow
mlflow run tests/resources/pip_project -e compute_intersection -P size=10000 --backend yarn

Example with conda project

  • Dependencies are pulled from conda.yaml
$ git clone https://github.com/criteo/mlflow-yarn
$ pip install mlflow
$ mlflow run tests/resources/conda_project -e compute_intersection -P size=10000 --backend yarn

More infos on setting up a project file.

Docker container environment is currently not supported.

About

Backend implementation for running MLFlow projects on Hadoop/YARN.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 98.4%
  • Other 1.6%