我有一个模型,我用管道注册了它:
register_step = PythonScriptStep(name = "Register Model",
source_directory = training_folder,
script_name = "register_model.py",
arguments = ['--model_folder', model_folder],
inputs=[model_folder],
compute_target = pipeline_cluster,
runconfig = pipeline_run_config,
allow_reuse = True)
Run Code Online (Sandbox Code Playgroud)
这是我的 register_model.py:
import argparse
import joblib
from azureml.core import Workspace, Model, Run
# Get parameters
parser = argparse.ArgumentParser()
parser.add_argument('--model_folder', type=str, dest='model_folder', default="model", help='model location')
args = parser.parse_args()
model_folder = args.model_folder
# Get the experiment run context
run = Run.get_context()
# Load the model
print("Loading model from " + model_folder)
model_file = model_folder + …Run Code Online (Sandbox Code Playgroud)