index out of range error when I use trained model to predict on another data set #95