Module: tff.simulation.baselines.stackoverflow  |  TensorFlow Federated (original) (raw)

Module: tff.simulation.baselines.stackoverflow

Stay organized with collections Save and categorize content based on your preferences.

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.

Other Members
DEFAULT_SEQUENCE_LENGTH 20
DEFAULT_TAG_VOCAB_SIZE 500
DEFAULT_WORD_VOCAB_SIZE 10000

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-09-20 UTC.