jprdonnelly/MachineLearningNotebooks
1
0
Fork 0
mirror of https://github.com/Azure/MachineLearningNotebooks.git synced 2025-12-19 17:17:04 -05:00
Code Issues Projects Releases Wiki Activity

update samples from Release-177 as a part of SDK release

Browse Source
This commit is contained in:
amlrelsa-ms
2023-04-17 10:22:59 +00:00
parent e4d84c8e45
commit 1178fcb0ba
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tutorials/machine-learning-pipelines-advanced/scripts/batch_scoring.py
View File

@@ -16,6 +16,7 @@ import tf_slim
from azureml.core import Run from azureml.core import Run
from azureml.core.model import Model from azureml.core.model import Model
from azureml.core.dataset import Dataset from azureml.core.dataset import Dataset
from tf_slim import nets
slim = tf_slim slim = tf_slim
@@ -41,13 +42,12 @@ def init():
parser.add_argument('--model_name', dest="model_name", required=True) parser.add_argument('--model_name', dest="model_name", required=True)
parser.add_argument('--labels_dir', dest="labels_dir", required=True) parser.add_argument('--labels_dir', dest="labels_dir", required=True)
args, _ = parser.parse_known_args() args, _ = parser.parse_known_args()
from nets import inception_v3, inception_utils
label_dict = get_class_label_dict(args.labels_dir) label_dict = get_class_label_dict(args.labels_dir)
classes_num = len(label_dict) classes_num = len(label_dict)
tf.disable_v2_behavior() tf.disable_v2_behavior()
with slim.arg_scope(inception_utils.inception_arg_scope()): with slim.arg_scope(nets.inception.inception_v3_arg_scope()):
input_images = tf.placeholder(tf.float32, [1, image_size, image_size, num_channel]) input_images = tf.placeholder(tf.float32, [1, image_size, image_size, num_channel])
logits, _ = inception_v3.inception_v3(input_images, logits, _ = nets.inception.inception_v3(input_images,
num_classes=classes_num, num_classes=classes_num,
is_training=False) is_training=False)
probabilities = tf.argmax(logits, 1) probabilities = tf.argmax(logits, 1)