Skip to content

Instantly share code, notes, and snippets.

@amogkam

amogkam/tune_mlflow_autolog.py

Created January 12, 2021 01:01
Show Gist options
  • Save amogkam/246889934f720b4ca7d405d181ec82b0 to your computer and use it in GitHub Desktop.
Save amogkam/246889934f720b4ca7d405d181ec82b0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tune_mlflow_autolog.py
from ray.tune.integration.mlflow import mlflow_mixin
@mlflow_mixin
def train_fn(config):
mlflow.autolog()
xgboost_results = xgb.train(config, ...)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment