How to use pd.get_dummies() with the test set