Missed TensorFlow World? Check out the recap. Learn more

tff.simulation.datasets.stackoverflow.load_word_counts

View source on GitHub

Loads the word counts for the Stackoverflow dataset.

tff.simulation.datasets.stackoverflow.load_word_counts(cache_dir=None)

Args:

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

Returns:

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.