amogkam/tune_mlflow_mixin.py

Created January 12, 2021 01:00

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/amogkam/b2e6418928353d5b6752bf8f7265ea71.js"></script>
Save amogkam/b2e6418928353d5b6752bf8f7265ea71 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

tune_mlflow_mixin.py

	from ray.tune.integration.mlflow import mlflow_mixin

	@mlflow_mixin
	def train_fn(config):
	...
	mlflow.log_metric(...)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment