Argmax and one-hot encoding of labels yet to be fixed: test_mlp