Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tff.simulation.datasets.stackoverflow.load_word_counts

View source

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.