in mnist-vscode-docs-sample/score.py [0:0]
def run(raw_data):
data = np.array(json.loads(raw_data)['data'])
# make prediction
out = output.eval(session = sess, feed_dict = {X: data})
y_hat = np.argmax(out, axis = 1)
return json.dumps(y_hat.tolist())