DrQAModel.ConfigΒΆ

Component: DrQAModel

class DrQAModel.Config[source]

Bases: BaseModel.Config

All Attributes (including base classes)

inputs: ModelInput = ModelInput()
dropout: float = 0.4
embedding: WordEmbedding.Config = WordEmbedding.Config(embed_dim=300, pretrained_embeddings_path='/mnt/vol/pytext/users/kushall/pretrained/glove.840B.300d.txt', vocab_from_pretrained_embeddings=True)
ques_rnn: StackedBidirectionalRNN.Config = StackedBidirectionalRNN.Config(dropout=0.4)
doc_rnn: StackedBidirectionalRNN.Config = StackedBidirectionalRNN.Config(dropout=0.4)
output_layer: SquadOutputLayer.Config = SquadOutputLayer.Config()
is_kd: bool = False

Default JSON

{
    "inputs": {
        "squad_input": {
            "SquadTensorizer": {
                "is_input": true,
                "column": "text",
                "tokenizer": {
                    "Tokenizer": {
                        "split_regex": "\\W+",
                        "lowercase": true,
                        "use_byte_offsets": false
                    }
                },
                "add_bos_token": false,
                "add_eos_token": false,
                "use_eos_token_for_bos": false,
                "max_seq_len": null,
                "vocab": {
                    "build_from_data": true,
                    "size_from_data": 0,
                    "min_counts": 0,
                    "vocab_files": []
                },
                "vocab_file_delimiter": " ",
                "doc_column": "doc",
                "ques_column": "question",
                "answers_column": "answers",
                "answer_starts_column": "answer_starts",
                "max_ques_seq_len": 64,
                "max_doc_seq_len": 256
            }
        },
        "has_answer": {
            "LabelTensorizer": {
                "is_input": false,
                "column": "has_answer",
                "allow_unknown": false,
                "pad_in_vocab": false,
                "label_vocab": null,
                "label_vocab_file": null,
                "add_labels": null
            }
        }
    },
    "dropout": 0.4,
    "embedding": {
        "load_path": null,
        "save_path": null,
        "freeze": false,
        "shared_module_key": null,
        "embed_dim": 300,
        "embedding_init_strategy": "random",
        "embedding_init_range": null,
        "embeddding_init_std": 0.02,
        "export_input_names": [
            "tokens_vals"
        ],
        "pretrained_embeddings_path": "/mnt/vol/pytext/users/kushall/pretrained/glove.840B.300d.txt",
        "vocab_file": "",
        "vocab_size": 0,
        "vocab_from_train_data": true,
        "vocab_from_all_data": false,
        "vocab_from_pretrained_embeddings": true,
        "lowercase_tokens": true,
        "min_freq": 1,
        "mlp_layer_dims": [],
        "padding_idx": null,
        "cpu_only": false,
        "skip_header": true,
        "delimiter": " "
    },
    "ques_rnn": {
        "load_path": null,
        "save_path": null,
        "freeze": false,
        "shared_module_key": null,
        "hidden_size": 32,
        "num_layers": 1,
        "dropout": 0.4,
        "bidirectional": true,
        "rnn_type": "lstm",
        "concat_layers": true
    },
    "doc_rnn": {
        "load_path": null,
        "save_path": null,
        "freeze": false,
        "shared_module_key": null,
        "hidden_size": 32,
        "num_layers": 1,
        "dropout": 0.4,
        "bidirectional": true,
        "rnn_type": "lstm",
        "concat_layers": true
    },
    "output_layer": {
        "load_path": null,
        "save_path": null,
        "freeze": false,
        "shared_module_key": null,
        "loss": {
            "CrossEntropyLoss": {}
        },
        "ignore_impossible": true,
        "pos_loss_weight": 0.5,
        "has_answer_loss_weight": 0.5,
        "false_label": "False",
        "max_answer_len": 30,
        "hard_weight": 0.0
    },
    "is_kd": false
}