-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathmain.py
More file actions
39 lines (33 loc) · 1.39 KB
/
main.py
File metadata and controls
39 lines (33 loc) · 1.39 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#from azureml import core
#from azureml.core import Workspace
import mlflow
import mlflow.sklearn
#workspace = Workspace.from_config()
#mlflow.set_tracking_uri(workspace.get_mlflow_tracking_uri())
mlflow.set_experiment("hyperparam")
config = {"COMPUTE": "cpu-cluster", "USE_CONDA": True}
uri = "MlflowProjectNotebooks/examples/hyperparam/"
"""
submitted_run = mlflow.projects.run(uri=uri,
backend = "azureml",
backend_config = config)
# train Keras DL model
submitted_run = mlflow.projects.run(uri=uri,
entry_point = "train",
backend = "azureml",
backend_config = config)
# Use hyperopt
submitted_run = mlflow.projects.run(uri=uri,
entry_point = "hyperopt",
backend = "azureml",
backend_config = config)
# Use GPyOpt
submitted_run = mlflow.projects.run(uri=uri,
entry_point = "gpyopt",
backend = "azureml",
backend_config = config)
"""
# Use random search
submitted_run = mlflow.projects.run(uri=uri,
entry_point = "random",
use_conda = False)