Module: tff.simulation.baselines.stackoverflow

Libraries for constructing baseline tasks for the Stack Overflow dataset.

Functions

create_tag_prediction_task(...): Creates a baseline task for tag prediction on Stack Overflow.

create_word_prediction_task(...): Creates a baseline task for next-word prediction on Stack Overflow.

DEFAULT_SEQUENCE_LENGTH 20
DEFAULT_TAG_VOCAB_SIZE 500
DEFAULT_WORD_VOCAB_SIZE 10000