tff.simulation.datasets.stackoverflow.load_word_counts

Loads the word counts for the Stack Overflow dataset.

cache_dir (Optional) directory to cache the downloaded file. If None, caches in Keras' default cache directory.

A collections.OrderedDict where the keys are string tokens, and the values are the counts of unique users who have at least one example in the training set containing that token in the body text.