load_or_cache_data doesn't surface enough options #12
Description
It loads only the test sets of [amazon, reddit, agnews], rather than the train or validation sets. It also doesn't allow you to subsample.