diff --git a/config.json b/config.json new file mode 100644 index 0000000..816bce1 --- /dev/null +++ b/config.json @@ -0,0 +1,59 @@ +{ + "architectures": [ + "BertForTokenClassification" + ], + "attention_probs_dropout_prob": 0.1, + "bos_token_id": null, + "do_sample": false, + "eos_token_ids": null, + "finetuning_task": null, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.1, + "hidden_size": 768, + "id2label": { + "0": "0", + "1": "VENDOR", + "2": "DATE", + "3": "HCW", + "4": "HOSPITAL", + "5": "ID", + "6": "PATIENT", + "7": "PHONE", + }, + "initializer_range": 0.02, + "intermediate_size": 3072, + "is_decoder": false, + "label2id": { + "0": 0, + "VENDOR": 1, + "DATE": 2, + "HCW": 3, + "HOSPITAL": 4, + "ID": 5, + "PATIENT": 6, + "PHONE": 7, + }, + "layer_norm_eps": 1e-12, + "length_penalty": 1.0, + "max_length": 20, + "max_position_embeddings": 512, + "model_type": "bert", + "num_attention_heads": 12, + "num_beams": 1, + "num_hidden_layers": 12, + "num_labels": 8, + "num_return_sequences": 1, + "output_attentions": false, + "output_hidden_states": false, + "output_past": true, + "pad_token_id": 0, + "pruned_heads": {}, + "repetition_penalty": 1.0, + "temperature": 1.0, + "top_k": 50, + "top_p": 1.0, + "torchscript": false, + "type_vocab_size": 2, + "use_bfloat16": false, + "vocab_size": 30522 +}