Skip to content

Instantly share code, notes, and snippets.

@Ben-Epstein

Ben-Epstein/Log MLFlow Info.py

Created November 5, 2020 15:32
Show Gist options
  • Save Ben-Epstein/6fafa85e363aeb952ba19ce46384c6ee to your computer and use it in GitHub Desktop.
Save Ben-Epstein/6fafa85e363aeb952ba19ce46384c6ee to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Log MLFlow Info.py
mlflow.log_pipeline_stages(model)
mlflow.log_feature_transformations(model)
mlflow.log_param('maxDepth', model.stages[-1].getOrDefault('maxDepth'))
mlflow.log_param('maxBins', model.stages[-1].getOrDefault('maxBins'))
mlflow.log_model(model, 'spark_dt') # Important!
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment