# Predict the categories of the test data
predicted_categories = model.predict(test_data.data)