asahi417's picture
add model
967c850
raw
history blame
13.2 kB
{
"_name_or_path": "cner_output/model/baseline/bert_base/best_model",
"architectures": [
"BertForTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"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": [
0.4588232636451721,
0.6268238425254822,
-0.8507410883903503,
-0.28580042719841003,
-0.5411331653594971,
1.2881312370300293,
1.0472862720489502,
-0.38454893231391907,
-1.284388542175293,
-0.8015868067741394,
0.15450958907604218,
-1.5412659645080566,
-0.5453099608421326,
-0.45127299427986145,
0.06227956339716911
],
"start_transitions": [
3.0626866817474365,
0.8901399374008179,
0.8567622303962708,
-1.9834086894989014,
0.7058414220809937,
-0.7659796476364136,
-0.987326979637146,
-1.3115990161895752,
-1.6038587093353271,
0.7949367761611938,
0.2939428389072418,
-0.9969339966773987,
1.2821195125579834,
-0.5328086018562317,
0.46629253029823303
],
"transitions": [
[
-0.06856999546289444,
0.36645984649658203,
-0.21410323679447174,
0.06793790310621262,
-0.15227515995502472,
0.1753452569246292,
0.09754128009080887,
-0.12793880701065063,
0.060049429535865784,
0.3986862599849701,
0.4137117564678192,
-0.022749241441488266,
0.09639288485050201,
0.7078870534896851,
-0.22414012253284454
],
[
0.08744336664676666,
0.23020756244659424,
0.17204752564430237,
-0.15548761188983917,
-0.251838743686676,
-0.48260292410850525,
0.13123716413974762,
-0.1390783041715622,
-0.03320292755961418,
0.2361810952425003,
0.3395829498767853,
-0.008724167011678219,
-0.20242442190647125,
-0.0422586090862751,
-0.1427241414785385
],
[
0.24087035655975342,
0.020674513652920723,
-0.007044103462249041,
0.465717077255249,
0.11266961693763733,
-0.22495201230049133,
0.09810738265514374,
-0.0787033811211586,
-0.01059836894273758,
0.04578001797199249,
0.12557506561279297,
0.005869352258741856,
0.29648515582084656,
0.19643500447273254,
0.405833899974823
],
[
-0.14240877330303192,
0.26585450768470764,
0.6220057010650635,
-0.37411966919898987,
-0.20534037053585052,
-0.025145141407847404,
-0.08551725000143051,
0.060617610812187195,
-0.051981810480356216,
-0.03395461663603783,
-0.3185342252254486,
-0.3586118519306183,
0.18121972680091858,
0.051030222326517105,
-0.1243407353758812
],
[
-0.009338058531284332,
-0.08902420848608017,
-0.13143935799598694,
-0.36317765712738037,
0.4261878430843353,
-0.07816917449235916,
0.09010732918977737,
0.045695845037698746,
0.1528170257806778,
0.01695443131029606,
0.13092565536499023,
0.12350313365459442,
-0.07707423716783524,
0.3080868422985077,
-0.025910988450050354
],
[
-0.009716416709125042,
0.06811506301164627,
-0.436259388923645,
0.5886178612709045,
-0.08773893117904663,
-0.13663692772388458,
-0.36365634202957153,
-0.15610387921333313,
-0.2890535295009613,
0.2886647880077362,
-0.11282126605510712,
-0.08684417605400085,
0.004607581067830324,
0.3027094602584839,
0.07905682176351547
],
[
-0.11060915887355804,
0.1953173726797104,
-0.02129223756492138,
-0.09051013737916946,
-0.08370254933834076,
-0.3755412995815277,
-0.24570095539093018,
0.0826478898525238,
-0.05054200813174248,
-0.21702523529529572,
-0.31449398398399353,
0.3117985427379608,
0.3654922544956207,
-0.31323695182800293,
0.18792077898979187
],
[
0.218822181224823,
-0.23124505579471588,
0.22463592886924744,
-0.13335612416267395,
0.7967448830604553,
-0.07453562319278717,
-0.18479441106319427,
0.4809357225894928,
0.2094232589006424,
-0.28631776571273804,
0.2572533190250397,
0.6722350120544434,
-0.47609663009643555,
-0.41936591267585754,
-0.0701839029788971
],
[
0.2784421443939209,
-0.1685822755098343,
-0.1094960868358612,
0.28578731417655945,
-0.07174302637577057,
0.5930020809173584,
-0.0864284336566925,
0.16199827194213867,
0.2849687933921814,
0.2531997561454773,
0.11696310341358185,
-0.36465322971343994,
0.4367746412754059,
0.24726858735084534,
-0.508527934551239
],
[
0.03771871328353882,
-0.09757845103740692,
0.32281672954559326,
-0.2882033884525299,
-0.3691042363643646,
0.12924355268478394,
-0.26587769389152527,
0.21671180427074432,
-0.11224760860204697,
-0.12496799975633621,
0.159481942653656,
0.5989466309547424,
-0.04836694523692131,
-0.05864289030432701,
-0.06600390374660492
],
[
0.04972785338759422,
0.11631549894809723,
0.02663993276655674,
0.08909134566783905,
0.02118838019669056,
0.6869929432868958,
-0.3074987530708313,
-0.021881429478526115,
0.34247836470603943,
-0.24236316978931427,
-0.1470607966184616,
-0.3854157328605652,
0.023870619013905525,
0.3785873055458069,
-0.35137078166007996
],
[
-0.17946091294288635,
-0.13627517223358154,
-0.0005373629974201322,
0.5927671790122986,
-0.1350403130054474,
0.3183661997318268,
0.1299688220024109,
-0.12246476858854294,
-0.14595994353294373,
0.23793143033981323,
0.8509373664855957,
0.3900677263736725,
0.27305108308792114,
0.32163798809051514,
-0.39687541127204895
],
[
0.1462329775094986,
-0.39886102080345154,
0.3025164306163788,
0.024430397897958755,
0.07892291992902756,
-0.7334970235824585,
0.12698613107204437,
0.3200628459453583,
-0.0814649760723114,
0.06127825006842613,
0.3119218349456787,
-0.23342493176460266,
0.11456283181905746,
0.12410871684551239,
0.14981859922409058
],
[
-0.2667269706726074,
-0.3662443161010742,
-0.0297089796513319,
0.14791344106197357,
-0.3424922525882721,
-0.29207950830459595,
0.4134122133255005,
0.5213165879249573,
-0.21123290061950684,
0.010724375024437904,
0.15926258265972137,
0.1943443864583969,
0.12391349673271179,
-0.4047512412071228,
-0.1096988320350647
],
[
0.10584335029125214,
-0.13750381767749786,
-0.35062938928604126,
-0.11933781951665878,
-0.041168272495269775,
0.20236517488956451,
-0.0396939292550087,
0.05537503585219383,
-0.2927897870540619,
-0.14394165575504303,
-0.24997498095035553,
-0.030565747991204262,
0.30675798654556274,
-0.03471777215600014,
0.06814266741275787
]
]
},
"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-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 0,
"position_embedding_type": "absolute",
"torch_dtype": "float32",
"transformers_version": "4.17.0",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 28996
}