test uses only kept columns from train after tfidf.