# define the test set
test_data = fetch_20newsgroups(subset="test", categories=text_categories)