Skip to content

Instantly share code, notes, and snippets.

@Paulescu

Paulescu/mlflow_setup.py

Created March 23, 2022 02:30
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Paulescu/921ab8f24c91072d952a916e46335e91 to your computer and use it in GitHub Desktop.
Save Paulescu/921ab8f24c91072d952a916e46335e91 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
mlflow_setup.py
## Paste the public URL generated by ngrok here
NGROK_URL = '<PLACEHOLDER>'
import mlflow
# connect mlflow client to the mlflow server that runs on localhost:5000
MLFLOW_SERVER_URI = NGROK_URL
mlflow.set_tracking_uri(str(MLFLOW_SERVER_URI))
EXPERIMENT_NAME = 'hyperparameter_search_from_colab'
mlflow.set_experiment(EXPERIMENT_NAME)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment