def do_inference_on_local_endpoint()

in tensorflow_script_mode_local_training_and_serving/tensorflow_script_mode_local_training_and_serving.py [0:0]

• 13 lines of code
• 3 McCabe index (conditional complexity)

def do_inference_on_local_endpoint(predictor, mode):
    print(f'\nStarting Inference on endpoint ({mode}).')
    correct_predictions = 0

    train_data = np.load('./data/train_data.npy')
    train_labels = np.load('./data/train_labels.npy')

    predictions = predictor.predict(train_data[:50])
    for i in range(0, 50):
        prediction = np.argmax(predictions['predictions'][i])
        label = train_labels[i]
        print('prediction is {}, label is {}, matched: {}'.format(prediction, label, prediction == label))
        if prediction == label:
            correct_predictions = correct_predictions + 1

    print('Calculated Accuracy from predictions: {}'.format(correct_predictions / 50))