jprdonnelly/MachineLearningNotebooks
1
0
Fork 0
mirror of https://github.com/Azure/MachineLearningNotebooks.git synced 2025-12-21 01:55:07 -05:00
Code Issues Projects Releases Wiki Activity

New SDK version

Browse Source
This commit is contained in:
Roope Astala
2018-11-05 15:27:36 -05:00
parent 32102e2aac
commit 7339c95ea0
46 changed files with 8200 additions and 4614 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
training/01.train-hyperparameter-tune-deploy-with-pytorch/pytorch_score.py
View File

@@ -53,5 +53,5 @@ def run(input_data):
pred_probs = softmax(model(img)).detach().numpy()[0]
index = torch.argmax(output, 1)
result = json.dumps({"label": classes[index], "probability": str(pred_probs[index])})
result = {"label": classes[index], "probability": str(pred_probs[index])}
return result