show_predict_digit(model_non_linear,
 test_dataset,
 0)