asahi417's picture
add model
17446cc
raw
history blame
13.3 kB
{
"_name_or_path": "cner_output/model/baseline_2021/roberta_base_dec2021/best_model",
"architectures": [
"RobertaForTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"bos_token_id": 0,
"classifier_dropout": null,
"crf_state_dict": {
"_constraint_mask": [
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
1.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
1.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
1.0,
0.0,
0.0
],
[
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0
]
],
"end_transitions": [
1.786523699760437,
-0.6999648213386536,
-0.7579379677772522,
-1.5308568477630615,
0.40813568234443665,
-0.9656944870948792,
-0.23096604645252228,
-0.7424056529998779,
0.06271817535161972,
-0.08851249516010284,
0.427242636680603,
0.8107076287269592,
0.75925213098526,
0.046232301741838455,
-0.9167554378509521
],
"start_transitions": [
-2.511308193206787,
-0.4623289704322815,
0.18997743725776672,
-0.13672193884849548,
-0.19599823653697968,
-0.09508302807807922,
-0.2742212116718292,
1.655328631401062,
0.2364160120487213,
-0.18163415789604187,
-0.5102927088737488,
2.0175106525421143,
0.8966974020004272,
-1.3882125616073608,
-0.12662316858768463
],
"transitions": [
[
-0.44308266043663025,
-0.09067360311746597,
0.02594163827598095,
0.054308146238327026,
-0.30705365538597107,
-0.2607330083847046,
0.2556297481060028,
0.3638714551925659,
-0.37692368030548096,
-0.14569033682346344,
0.17723748087882996,
0.39900416135787964,
-0.12451571971178055,
0.035905562341213226,
-0.35420721769332886
],
[
0.07558207213878632,
-0.04898068308830261,
-0.3468376696109772,
0.0819142609834671,
-0.24118109047412872,
-0.16222117841243744,
0.059606537222862244,
0.579510509967804,
0.12647897005081177,
0.2972663342952728,
-0.15214888751506805,
0.07717656344175339,
-0.20168255269527435,
-0.10988985747098923,
-0.15759176015853882
],
[
0.18069757521152496,
0.24784591794013977,
-0.1826421022415161,
0.11285563558340073,
0.029766272753477097,
0.1829925775527954,
0.020348960533738136,
-0.5790755748748779,
0.3058984875679016,
0.3696746826171875,
0.2943328022956848,
0.4187123775482178,
-0.06584056466817856,
0.5778209567070007,
-0.1550922840833664
],
[
-0.17005695402622223,
0.21888934075832367,
-0.0987241268157959,
-0.2138545662164688,
-0.2961842715740204,
0.21853631734848022,
0.08733738958835602,
0.1732124537229538,
0.012827172875404358,
-0.39205366373062134,
-0.2397550791501999,
0.7359107732772827,
-0.30322200059890747,
0.30835142731666565,
0.18199004232883453
],
[
-0.23071691393852234,
0.07519257813692093,
-0.3006647825241089,
-0.22338630259037018,
0.36292341351509094,
-0.1669275015592575,
-0.10870001465082169,
-0.417815238237381,
0.33958712220191956,
-0.08915771543979645,
0.16988518834114075,
-0.13071578741073608,
-0.13169747591018677,
-0.1825239509344101,
-0.06032129004597664
],
[
-0.11419851332902908,
0.11649321764707565,
-0.15078102052211761,
0.05721569433808327,
0.15539152920246124,
0.1490965634584427,
-0.06250990927219391,
0.079411581158638,
0.268951416015625,
0.2026214301586151,
0.3623451292514801,
0.6320552825927734,
-0.29348284006118774,
-0.3976886570453644,
-0.11971747875213623
],
[
0.5550814270973206,
-0.20710957050323486,
0.26982447504997253,
0.1094808280467987,
0.4171890914440155,
0.2847338318824768,
-0.29747381806373596,
0.11556798964738846,
-0.42085689306259155,
-0.20381823182106018,
-0.1320376694202423,
-0.08456902205944061,
0.43453216552734375,
-0.03658383712172508,
-0.012955465354025364
],
[
0.052638061344623566,
0.1243215873837471,
-0.10308931022882462,
0.4010932445526123,
-0.23081545531749725,
0.10897678136825562,
-0.30448317527770996,
0.37836843729019165,
-0.13308532536029816,
-0.22834356129169464,
-0.32330456376075745,
0.18331480026245117,
-0.07475696504116058,
0.4401257038116455,
0.1768082082271576
],
[
-0.30624452233314514,
-0.4645151197910309,
0.12465409189462662,
0.16773360967636108,
-0.014425227418541908,
0.07834259420633316,
-0.0881722941994667,
-0.1853989064693451,
0.019380783662199974,
-0.058270394802093506,
-0.034863103181123734,
-0.31407323479652405,
-0.13009674847126007,
-0.37027472257614136,
0.32549360394477844
],
[
0.2504803538322449,
0.19011732935905457,
0.24709364771842957,
-0.22658921778202057,
-0.07339753210544586,
0.1528608500957489,
-0.36086344718933105,
0.18543387949466705,
-0.13220855593681335,
-0.16181913018226624,
-0.573129415512085,
0.029792217537760735,
0.4576672315597534,
-0.14162465929985046,
0.0316760390996933
],
[
0.20881691575050354,
0.3950963616371155,
-0.14713308215141296,
-0.21527227759361267,
0.16928252577781677,
0.22524012625217438,
-0.620922327041626,
0.17128828167915344,
0.13766540586948395,
-0.31718698143959045,
-0.106533482670784,
0.050578828901052475,
0.024287668988108635,
-0.5655285120010376,
-0.16035562753677368
],
[
-0.31755414605140686,
-0.07679922133684158,
-0.15979549288749695,
0.017346711829304695,
0.1461765319108963,
0.037325579673051834,
0.03056003898382187,
-0.14054889976978302,
0.28088274598121643,
-0.7118891477584839,
0.32795894145965576,
0.43596091866493225,
0.08656708151102066,
0.22009947896003723,
0.32324105501174927
],
[
-0.08870566636323929,
-0.04523065313696861,
-0.00709886709228158,
0.25830698013305664,
-0.03580803424119949,
0.05059157684445381,
0.08641693741083145,
0.630769670009613,
-0.09486010670661926,
0.3704228103160858,
-0.42412737011909485,
0.36427462100982666,
0.2742330729961395,
0.16411158442497253,
0.33911505341529846
],
[
-0.47170379757881165,
0.1916237622499466,
0.4052870273590088,
0.002804133575409651,
-0.001292021363042295,
0.059521690011024475,
-0.019467193633317947,
0.29833313822746277,
0.05193063244223595,
0.2820134460926056,
-0.07352016866207123,
0.12595157325267792,
0.061767756938934326,
0.11658620834350586,
0.27741602063179016
],
[
-0.052867282181978226,
-0.30663543939590454,
0.29321423172950745,
-0.34890830516815186,
-0.11623632162809372,
-0.03872501850128174,
0.127924382686615,
0.2524586617946625,
-0.4238189458847046,
0.03921777382493019,
-0.03415319323539734,
0.3902970850467682,
-0.05142243579030037,
0.10664427280426025,
0.45849040150642395
]
]
},
"eos_token_id": 2,
"gradient_checkpointing": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "B-corporation",
"1": "B-creative_work",
"2": "B-event",
"3": "B-group",
"4": "B-location",
"5": "B-person",
"6": "B-product",
"7": "I-corporation",
"8": "I-creative_work",
"9": "I-event",
"10": "I-group",
"11": "I-location",
"12": "I-person",
"13": "I-product",
"14": "O"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"B-corporation": 0,
"B-creative_work": 1,
"B-event": 2,
"B-group": 3,
"B-location": 4,
"B-person": 5,
"B-product": 6,
"I-corporation": 7,
"I-creative_work": 8,
"I-event": 9,
"I-group": 10,
"I-location": 11,
"I-person": 12,
"I-product": 13,
"O": 14
},
"layer_norm_eps": 1e-05,
"max_position_embeddings": 514,
"model_type": "roberta",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 1,
"position_embedding_type": "absolute",
"torch_dtype": "float32",
"transformers_version": "4.17.0",
"type_vocab_size": 1,
"use_cache": true,
"vocab_size": 50265
}