flambe.nlp.transformers

Package Contents

class flambe.nlp.transformers.BertTextField[source]

Bases: flambe.nlp.transformers.utils.TransformerTextField

Integrate the pytorch_transformers BertTokenizer.

Currently available aliases:
. bert-base-uncased . bert-large-uncased . bert-base-cased . bert-large-cased . bert-base-multilingual-uncased . bert-base-multilingual-cased . bert-base-chinese . bert-base-german-cased . bert-large-uncased-whole-word-masking . bert-large-cased-whole-word-masking . bert-large-uncased-whole-word-masking-finetuned-squad . bert-large-cased-whole-word-masking-finetuned-squad . bert-base-cased-finetuned-mrpc
_cls
class flambe.nlp.transformers.BertEmbedder[source]

Bases: flambe.nlp.transformers.utils.TransformerEmbedder

Integrate the pytorch_transformers BertModel.

Currently available aliases:
. bert-base-uncased . bert-large-uncased . bert-base-cased . bert-large-cased . bert-base-multilingual-uncased . bert-base-multilingual-cased . bert-base-chinese . bert-base-german-cased . bert-large-uncased-whole-word-masking . bert-large-cased-whole-word-masking . bert-large-uncased-whole-word-masking-finetuned-squad . bert-large-cased-whole-word-masking-finetuned-squad . bert-base-cased-finetuned-mrpc
_cls
class flambe.nlp.transformers.RobertaTextField[source]

Bases: flambe.nlp.transformers.utils.TransformerTextField

Integrate the pytorch_transformers RobertaTokenizer.

Currently available aliases:
. roberta-base . roberta-large . roberta-large-mnli
_cls
class flambe.nlp.transformers.RobertaEmbedder[source]

Bases: flambe.nlp.transformers.utils.TransformerEmbedder

Integrate the pytorch_transformers RobertaModel.

Currently available aliases:
. roberta-base . roberta-large . roberta-large-mnli
_cls
class flambe.nlp.transformers.GPTTextField[source]

Bases: flambe.nlp.transformers.utils.TransformerTextField

Integrate the pytorch_transformers OpenAIGPTTokenizer.

Currently available aliases:
. openai-gpt
_cls
class flambe.nlp.transformers.GPTEmbedder[source]

Bases: flambe.nlp.transformers.utils.TransformerEmbedder

Integrate the pytorch_transformers OpenAIGPTmodel.

Currently available aliases:
. openai-gpt
_cls
class flambe.nlp.transformers.GPT2TextField[source]

Bases: flambe.nlp.transformers.utils.TransformerTextField

Integrate the pytorch_transformers GPT2Tokenizer.

Currently available aliases:
. gpt2 . gpt2-medium . gpt2-large
_cls
class flambe.nlp.transformers.GPT2Embedder[source]

Bases: flambe.nlp.transformers.utils.TransformerEmbedder

Integrate the pytorch_transformers GPT2Model.

Currently available aliases:
. gpt2 . gpt2-medium . gpt2-large
_cls
class flambe.nlp.transformers.XLMTextField[source]

Bases: flambe.nlp.transformers.utils.TransformerTextField

Integrate the pytorch_transformers XLMTokenizer.

Currently available aliases:
. xlm-mlm-en-2048 . xlm-mlm-ende-1024 . xlm-mlm-enfr-1024 . xlm-mlm-enro-1024 . xlm-mlm-tlm-xnli15-1024 . xlm-mlm-xnli15-1024 . xlm-clm-enfr-1024 . xlm-clm-ende-1024
_cls
class flambe.nlp.transformers.XLMEmbedder[source]

Bases: flambe.nlp.transformers.utils.TransformerEmbedder

Integrate the pytorch_transformers XLMModel.

Currently available aliases:
. xlm-mlm-en-2048 . xlm-mlm-ende-1024 . xlm-mlm-enfr-1024 . xlm-mlm-enro-1024 . xlm-mlm-tlm-xnli15-1024 . xlm-mlm-xnli15-1024 . xlm-clm-enfr-1024 . xlm-clm-ende-1024
_cls
class flambe.nlp.transformers.XLNetTextField[source]

Bases: flambe.nlp.transformers.utils.TransformerTextField

Integrate the pytorch_transformers XLNetTokenizer.

Currently available aliases:
. xlnet-base-cased . xlnet-large-cased
_cls
class flambe.nlp.transformers.XLNetEmbedder[source]

Bases: flambe.nlp.transformers.utils.TransformerEmbedder

Integrate the pytorch_transformers XLNetModel.

Currently available aliases:
. xlnet-base-cased . xlnet-large-cased
_cls
class flambe.nlp.transformers.TransfoXLTextField[source]

Bases: flambe.nlp.transformers.utils.TransformerTextField

Integrate the pytorch_transformers TransfoXLTokenizer.

Currently available aliases:
. transfo-xl-wt103
_cls
class flambe.nlp.transformers.TransfoXLEmbedder[source]

Bases: flambe.nlp.transformers.utils.TransformerEmbedder

Integrate the pytorch_transformers TransfoXLModel.

Currently available aliases:
. transfo-xl-wt103
_cls
class flambe.nlp.transformers.AdamW[source]

Bases: flambe.Component, pytorch_transformers.AdamW

class flambe.nlp.transformers.ConstantLRSchedule[source]

Bases: flambe.Component, pytorch_transformers.ConstantLRSchedule

class flambe.nlp.transformers.WarmupConstantSchedule[source]

Bases: flambe.Component, pytorch_transformers.WarmupConstantSchedule

class flambe.nlp.transformers.WarmupLinearSchedule[source]

Bases: flambe.Component, pytorch_transformers.WarmupLinearSchedule