Attend the Women in ML Symposium on December 7 Register now

Module: tff.simulation.datasets.stackoverflow

Libraries for the Stackoverflow dataset for federated learning simulation.

Functions

create_synthetic_data_dictionary(...)

get_synthetic(...): Returns a small synthetic dataset for testing.

get_synthetic_tag_counts(...): Returns a small dictionary of tag counts.

get_synthetic_word_counts(...): Returns a small dictionary of word counts.

load_data(...): Loads the federated Stack Overflow dataset.

load_tag_counts(...): Loads the tag counts for the Stack Overflow dataset.

load_word_counts(...): Loads the word counts for the Stack Overflow dataset.