Typo: train_outputs vs test_outputs.