This repository was archived by the owner on Sep 25, 2026. It is now read-only.
Repository navigation
Expand file tree
/
Copy pathevaluate_model.py
More file actions
160 lines (144 loc) · 5.88 KB
/
Copy pathevaluate_model.py
File metadata and controls
160 lines (144 loc) · 5.88 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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
"""
Copyright (C) Microsoft Corporation. All rights reserved.
Microsoft Corporation (“Microsoft”) grants you a nonexclusive, perpetual,
royalty-free right to use, copy, and modify the software code provided by us
("Software Code"). You may not sublicense the Software Code or any use of it
(except to your affiliates and to vendors to perform work on your behalf)
through distribution, network access, service agreement, lease, rental, or
otherwise. This license does not purport to express any claim of ownership over
data you may have shared with Microsoft in the creation of the Software Code.
Unless applicable law gives you more rights, Microsoft reserves all other
rights not expressly granted herein, whether by implication, estoppel or
otherwise.
THE SOFTWARE CODE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS
OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
MICROSOFT OR ITS LICENSORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ARISING IN ANY WAY OUT OF THE USE OF THE SOFTWARE CODE, EVEN IF ADVISED OF THE
POSSIBILITY OF SUCH DAMAGE.
"""
from azureml.core import Run
import argparse
import traceback
from diabetes_regression.util.model_helper import get_model
from utils.logger.logger_interface import Severity
from utils.logger.observability import Observability
observability = Observability()
run = Run.get_context()
# if you would like to run this script on a local computer
# the following code is a good starting point for you
# use
# python -m evaluate.evaluate_model
# in diabetes_regression folder context
# if (run.id.startswith('OfflineRun')):
# from dotenv import load_dotenv
# # For local development, set values in this section
# load_dotenv()
# sources_dir = os.environ.get("SOURCES_DIR_TRAIN")
# if (sources_dir is None):
# sources_dir = '.'
# path_to_util = os.path.join(".", sources_dir, "util")
# sys.path.append(os.path.abspath(path_to_util)) # NOQA: E402
# from model_helper import get_model
# workspace_name = os.environ.get("WORKSPACE_NAME")
# experiment_name = os.environ.get("EXPERIMENT_NAME")
# resource_group = os.environ.get("RESOURCE_GROUP")
# subscription_id = os.environ.get("SUBSCRIPTION_ID")
# tenant_id = os.environ.get("TENANT_ID")
# model_name = os.environ.get("MODEL_NAME")
# app_id = os.environ.get('SP_APP_ID')
# app_secret = os.environ.get('SP_APP_SECRET')
# build_id = os.environ.get('BUILD_BUILDID')
# # run_id useful to query previous runs
# run_id = "57fee47f-5ae8-441c-bc0c-d4c371f32d70"
# aml_workspace = Workspace.get(
# name=workspace_name,
# subscription_id=subscription_id,
# resource_group=resource_group
# )
# ws = aml_workspace
# exp = Experiment(ws, experiment_name)
# comment the following three lines
# if you would like to use Offline mode
exp = run.experiment
ws = run.experiment.workspace
run_id = 'amlcompute'
parser = argparse.ArgumentParser("evaluate")
parser.add_argument(
"--run_id",
type=str,
help="Training run ID",
)
parser.add_argument(
"--model_name",
type=str,
help="Name of the Model",
default="diabetes_model.pkl",
)
parser.add_argument(
"--allow_run_cancel",
type=str,
help="Set this to false to avoid evaluation step from cancelling "
"run after an unsuccessful evaluation",
default="true",
)
args = parser.parse_args()
if (args.run_id is not None):
run_id = args.run_id
if (run_id == 'amlcompute'):
run_id = run.parent.id
model_name = args.model_name
metric_eval = "mse"
allow_run_cancel = args.allow_run_cancel
# Parameterize the matrices on which the models should be compared
# Add golden data set on which all the model performance can be evaluated
try:
firstRegistration = False
tag_name = 'experiment_name'
model = get_model(
model_name=model_name,
tag_name=tag_name,
tag_value=exp.name,
aml_workspace=ws)
if (model is not None):
production_model_mse = 10000
if (metric_eval in model.tags):
production_model_mse = float(model.tags[metric_eval])
new_model_mse = float(run.parent.get_metrics().get(metric_eval))
if (production_model_mse is None or new_model_mse is None):
observability.log("Unable to find" +
metric_eval + "metrics, exiting evaluation")
if ((allow_run_cancel).lower() == 'true'):
run.parent.cancel()
else:
observability.log(
"Current Production model mse: {}, "
"New trained model mse: {}".format(
production_model_mse, new_model_mse
)
)
if (new_model_mse < production_model_mse):
observability.log("New trained model performs better, "
"thus it should be registered")
else:
observability.log("New trained model metric is worse "
"than or equal to "
"production model so skipping "
"model registration.")
if ((allow_run_cancel).lower() == 'true'):
run.parent.cancel()
else:
observability.log("This is the first model, "
"thus it should be registered")
except Exception:
traceback.print_exc(limit=None, file=None, chain=True)
observability.log(
description="Something went wrong trying to evaluate. Exiting.",
severity=Severity.ERROR)
raise