Have a question? Connect with the community at the TensorFlow Forum Visit Forum

Module: tfnlp.networks

Networks package definition.

Classes

class AlbertEncoder: ALBERT (https://arxiv.org/abs/1810.04805) text encoder network.

class BertEncoder: Bi-directional Transformer-based encoder network.

class Classification: Classification network head for BERT modeling.

class EncoderScaffold: Bi-directional Transformer-based encoder network scaffold.

class MobileBERTEncoder: A Keras functional API implementation for MobileBERT encoder.

class PackedSequenceEmbedding: An embedding network supporting packed sequences and position ids.

class SpanLabeling: Span labeling network head for BERT modeling.

class TransformerEncoder: Bi-directional Transformer-based encoder network.

class XLNetBase: Base XLNet model.

class XLNetSpanLabeling: Span labeling network head for XLNet on SQuAD2.0.