ModelInputConfigΒΆ

Component: Module

class pytext.config.doc_classification.ModelInputConfig[source]

Bases: Module.Config

All Attributes (including base classes)

load_path: Optional[str] = None
save_path: Optional[str] = None
freeze: bool = False
shared_module_key: Optional[str] = None
word_feat: WordEmbedding.Config = WordEmbedding.Config()
dict_feat: Optional[DictEmbedding.Config] = None
char_feat: Optional[CharacterEmbedding.Config] = None
pretrained_model_embedding: Optional[PretrainedModelEmbedding.Config] = None
dense_feat: Optional[FloatVectorConfig] = None

Default JSON

{
    "load_path": null,
    "save_path": null,
    "freeze": false,
    "shared_module_key": null,
    "word_feat": {
        "embed_dim": 100,
        "freeze": false,
        "embedding_init_strategy": "random",
        "embedding_init_range": null,
        "export_input_names": [
            "tokens_vals"
        ],
        "pretrained_embeddings_path": "",
        "vocab_file": "",
        "vocab_size": 0,
        "vocab_from_train_data": true,
        "vocab_from_all_data": false,
        "vocab_from_pretrained_embeddings": false,
        "lowercase_tokens": true,
        "min_freq": 1,
        "mlp_layer_dims": []
    },
    "dict_feat": null,
    "char_feat": null,
    "pretrained_model_embedding": null,
    "dense_feat": null
}