Yes, the input layer needs to be size 49. The output layer needs to be size 1. Then it'll work for both the train and test data.