pvcastro's picture
Upload BertForEmbedderCRFTokenClassification
5a4607f verified
raw
history blame
4.72 kB
{
"_name_or_path": "/media/pedro/repositorios/models/datalawyer/ner/pedidos-transformers-crf-fp16-v5.3",
"architectures": [
"BertForEmbedderCRFTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_dropout": 0.2,
"constrain_crf": true,
"directionality": "bidi",
"dropout_task": 0.2,
"finetuning_task": "ner",
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "B-ATRIBUICAO",
"1": "I-ATRIBUICAO",
"2": "E-ATRIBUICAO",
"3": "S-ATRIBUICAO",
"4": "B-DECISAO",
"5": "I-DECISAO",
"6": "E-DECISAO",
"7": "S-DECISAO",
"8": "B-FUNCAO",
"9": "I-FUNCAO",
"10": "E-FUNCAO",
"11": "S-FUNCAO",
"12": "B-FUNDAMENTO",
"13": "I-FUNDAMENTO",
"14": "E-FUNDAMENTO",
"15": "S-FUNDAMENTO",
"16": "B-LOCAL",
"17": "I-LOCAL",
"18": "E-LOCAL",
"19": "S-LOCAL",
"20": "B-ORGANIZACAO",
"21": "I-ORGANIZACAO",
"22": "E-ORGANIZACAO",
"23": "S-ORGANIZACAO",
"24": "B-PEDIDO",
"25": "I-PEDIDO",
"26": "E-PEDIDO",
"27": "S-PEDIDO",
"28": "B-PESSOA",
"29": "I-PESSOA",
"30": "E-PESSOA",
"31": "S-PESSOA",
"32": "B-REFLEXO",
"33": "I-REFLEXO",
"34": "E-REFLEXO",
"35": "S-REFLEXO",
"36": "B-TIPO_ACAO",
"37": "I-TIPO_ACAO",
"38": "E-TIPO_ACAO",
"39": "S-TIPO_ACAO",
"40": "B-TRIBUNAL",
"41": "I-TRIBUNAL",
"42": "E-TRIBUNAL",
"43": "S-TRIBUNAL",
"44": "B-VALOR_ACORDO",
"45": "I-VALOR_ACORDO",
"46": "E-VALOR_ACORDO",
"47": "S-VALOR_ACORDO",
"48": "B-VALOR_CAUSA",
"49": "I-VALOR_CAUSA",
"50": "E-VALOR_CAUSA",
"51": "S-VALOR_CAUSA",
"52": "B-VALOR_CONDENACAO",
"53": "I-VALOR_CONDENACAO",
"54": "E-VALOR_CONDENACAO",
"55": "S-VALOR_CONDENACAO",
"56": "B-VALOR_CUSTAS",
"57": "I-VALOR_CUSTAS",
"58": "E-VALOR_CUSTAS",
"59": "S-VALOR_CUSTAS",
"60": "B-VALOR_PEDIDO",
"61": "I-VALOR_PEDIDO",
"62": "E-VALOR_PEDIDO",
"63": "S-VALOR_PEDIDO",
"64": "B-VARA",
"65": "I-VARA",
"66": "E-VARA",
"67": "S-VARA",
"68": "O"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"B-ATRIBUICAO": 0,
"B-DECISAO": 4,
"B-FUNCAO": 8,
"B-FUNDAMENTO": 12,
"B-LOCAL": 16,
"B-ORGANIZACAO": 20,
"B-PEDIDO": 24,
"B-PESSOA": 28,
"B-REFLEXO": 32,
"B-TIPO_ACAO": 36,
"B-TRIBUNAL": 40,
"B-VALOR_ACORDO": 44,
"B-VALOR_CAUSA": 48,
"B-VALOR_CONDENACAO": 52,
"B-VALOR_CUSTAS": 56,
"B-VALOR_PEDIDO": 60,
"B-VARA": 64,
"E-ATRIBUICAO": 2,
"E-DECISAO": 6,
"E-FUNCAO": 10,
"E-FUNDAMENTO": 14,
"E-LOCAL": 18,
"E-ORGANIZACAO": 22,
"E-PEDIDO": 26,
"E-PESSOA": 30,
"E-REFLEXO": 34,
"E-TIPO_ACAO": 38,
"E-TRIBUNAL": 42,
"E-VALOR_ACORDO": 46,
"E-VALOR_CAUSA": 50,
"E-VALOR_CONDENACAO": 54,
"E-VALOR_CUSTAS": 58,
"E-VALOR_PEDIDO": 62,
"E-VARA": 66,
"I-ATRIBUICAO": 1,
"I-DECISAO": 5,
"I-FUNCAO": 9,
"I-FUNDAMENTO": 13,
"I-LOCAL": 17,
"I-ORGANIZACAO": 21,
"I-PEDIDO": 25,
"I-PESSOA": 29,
"I-REFLEXO": 33,
"I-TIPO_ACAO": 37,
"I-TRIBUNAL": 41,
"I-VALOR_ACORDO": 45,
"I-VALOR_CAUSA": 49,
"I-VALOR_CONDENACAO": 53,
"I-VALOR_CUSTAS": 57,
"I-VALOR_PEDIDO": 61,
"I-VARA": 65,
"O": 68,
"S-ATRIBUICAO": 3,
"S-DECISAO": 7,
"S-FUNCAO": 11,
"S-FUNDAMENTO": 15,
"S-LOCAL": 19,
"S-ORGANIZACAO": 23,
"S-PEDIDO": 27,
"S-PESSOA": 31,
"S-REFLEXO": 35,
"S-TIPO_ACAO": 39,
"S-TRIBUNAL": 43,
"S-VALOR_ACORDO": 47,
"S-VALOR_CAUSA": 51,
"S-VALOR_CONDENACAO": 55,
"S-VALOR_CUSTAS": 59,
"S-VALOR_PEDIDO": 63,
"S-VARA": 67
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"output_past": true,
"pad_token_id": 0,
"pooler_fc_size": 768,
"pooler_num_attention_heads": 12,
"pooler_num_fc_layers": 3,
"pooler_size_per_head": 128,
"pooler_type": "first_token_transform",
"position_embedding_type": "absolute",
"quantization_config": {
"bnb_4bit_compute_dtype": "float32",
"bnb_4bit_quant_type": "fp4",
"bnb_4bit_use_double_quant": false,
"llm_int8_enable_fp32_cpu_offload": false,
"llm_int8_has_fp16_weight": false,
"llm_int8_skip_modules": null,
"llm_int8_threshold": 6.0,
"load_in_4bit": false,
"load_in_8bit": true,
"quant_method": "bitsandbytes"
},
"torch_dtype": "float16",
"transformers_version": "4.36.2",
"type_vocab_size": 2,
"use_cache": true,
"use_crf": true,
"vocab_size": 29794
}