BertModelInputΒΆ
-
class
pytext.models.bert_classification_models.
BertModelInput
Bases:
ModelInput
All Attributes (including base classes)
- tokens: BERTTensorizer.Config = BERTTensorizer.Config(max_seq_len=
128
)- dense: Optional[FloatListTensorizer.Config] =
None
- labels: LabelTensorizer.Config = LabelTensorizer.Config()
- num_tokens: NtokensTensorizer.Config = NtokensTensorizer.Config(names=
['tokens']
, indexes=[2]
)
Default JSON
{
"tokens": {
"BERTTensorizer": {
"is_input": true,
"columns": [
"text"
],
"tokenizer": {
"WordPieceTokenizer": {
"basic_tokenizer": {
"split_regex": "\\s+",
"lowercase": true
},
"wordpiece_vocab_path": "/mnt/vol/nlp_technologies/bert/uncased_L-12_H-768_A-12/vocab.txt"
}
},
"base_tokenizer": null,
"vocab_file": "/mnt/vol/nlp_technologies/bert/uncased_L-12_H-768_A-12/vocab.txt",
"max_seq_len": 128
}
},
"dense": null,
"labels": {
"LabelTensorizer": {
"is_input": false,
"column": "label",
"allow_unknown": false,
"pad_in_vocab": false,
"label_vocab": null
}
},
"num_tokens": {
"is_input": false,
"names": [
"tokens"
],
"indexes": [
2
]
}
}