Fixed indexOutOfBounds when using feature selection: have to filter out features in the test set too.