Hugging Face's logo

tner
/
roberta-large-mit-restaurant

Token Classification
Transformers
PyTorch
roberta
Eval Results
Inference Endpoints
Model card Files Community
2
roberta-large-mit-restaurant / config.json
asahi417's picture
asahi417
model update
75c8d1e
raw
history blame
16.3 kB
{
"_name_or_path": "tner_ckpt/mit_restaurant_roberta_large/model_rcsnba/epoch_5",
"architectures": [
"RobertaForTokenClassification"
],
"attention_probs_dropout_prob": 0.1,
"bos_token_id": 0,
"classifier_dropout": null,
"crf_state_dict": {
"_constraint_mask": [
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
1.0,
1.0,
0.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
1.0,
1.0,
0.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0
],
[
1.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
0.0,
1.0,
1.0,
0.0,
1.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,
0.0,
0.0,
0.0,
0.0
]
],
"end_transitions": [
0.7360687255859375,
-1.031581163406372,
0.9866265058517456,
1.9954051971435547,
-0.21944227814674377,
0.9656914472579956,
-0.635370135307312,
0.5445006489753723,
0.27633094787597656,
-0.005909058731049299,
-0.7182324528694153,
0.5392410755157471,
-0.7462356686592102,
-2.2462260723114014,
1.0836721658706665,
0.04674846678972244,
0.18168555200099945
],
"start_transitions": [
0.11282581090927124,
-0.5175673365592957,
-0.8899797201156616,
0.18878838419914246,
0.22193501889705658,
0.42420968413352966,
-0.11027862876653671,
0.7695011496543884,
-0.3819623589515686,
0.05800779163837433,
-0.7716839909553528,
-1.3946560621261597,
-2.0551090240478516,
0.6582558751106262,
-0.9261692762374878,
-0.6792001128196716,
0.058188848197460175
],
"transitions": [
[
-0.15182898938655853,
-0.3338739275932312,
-0.4415439963340759,
0.08789703249931335,
0.08695892989635468,
-0.1922253668308258,
-0.1455051153898239,
-0.20090816915035248,
-0.0287786778062582,
-0.2247808873653412,
-0.16542282700538635,
0.044164374470710754,
0.24052512645721436,
-0.12844151258468628,
0.14698180556297302,
-0.49263817071914673,
0.17261715233325958
],
[
0.37535911798477173,
0.19647495448589325,
0.36591529846191406,
0.1289680302143097,
-0.054969288408756256,
-0.11405882984399796,
0.01863745041191578,
0.1843152642250061,
0.06600829213857651,
0.7022599577903748,
0.19833503663539886,
0.3102588355541229,
0.19278442859649658,
0.3879980742931366,
-0.17290253937244415,
0.5220606327056885,
0.17804183065891266
],
[
0.11496956646442413,
-0.02462540566921234,
-0.28180068731307983,
-0.10623500496149063,
-0.07495591044425964,
0.04812565818428993,
0.13284999132156372,
0.17153944075107574,
0.01631355658173561,
0.44744062423706055,
-0.04717656224966049,
0.2213021218776703,
0.26431015133857727,
-0.24772833287715912,
-0.19885113835334778,
0.02920069731771946,
0.35805606842041016
],
[
-0.06933118402957916,
0.15567733347415924,
0.253507137298584,
0.09900859743356705,
0.08491659164428711,
0.11925014853477478,
-0.2230781465768814,
0.0833144560456276,
0.014857443980872631,
-0.48656606674194336,
-0.21304617822170258,
-0.1804332286119461,
-0.6107743978500366,
0.07936467975378036,
0.26377078890800476,
0.15834248065948486,
0.00647344347089529
],
[
-0.49816271662712097,
-0.27096712589263916,
-0.09031641483306885,
0.23066899180412292,
0.3341634273529053,
0.24149982631206512,
0.9208809733390808,
0.35747119784355164,
0.12701252102851868,
0.457673043012619,
-0.1881372183561325,
0.2987460792064667,
0.05171642079949379,
-0.2482197880744934,
0.09817705303430557,
-0.08112853765487671,
-0.059872500598430634
],
[
0.19921885430812836,
0.31734371185302734,
0.3676517903804779,
0.3893914222717285,
-0.50760418176651,
0.023447703570127487,
0.038262613117694855,
-0.42159387469291687,
0.2511283755302429,
-0.2948586940765381,
-0.30983737111091614,
-0.042588550597429276,
0.1021285429596901,
-0.08279505372047424,
0.20223405957221985,
0.15694308280944824,
0.2092602550983429
],
[
0.08988863229751587,
-0.11746697127819061,
-0.13020983338356018,
-0.028597405180335045,
-0.08930719643831253,
0.48039567470550537,
0.3854551315307617,
0.11058463901281357,
0.33238568902015686,
-0.08155417442321777,
0.042421407997608185,
0.012302840128540993,
-0.25249117612838745,
0.24550507962703705,
-0.08428964018821716,
0.014765261672437191,
0.045157257467508316
],
[
0.2325226068496704,
-0.03915967419743538,
-0.3954221308231354,
0.11455757170915604,
-0.3053692877292633,
-0.16449660062789917,
0.2053939551115036,
-0.4745903015136719,
0.4925399124622345,
-0.25869667530059814,
-0.04963960498571396,
0.029781360179185867,
-0.31160813570022583,
-0.12200227379798889,
0.046613309532403946,
0.24962788820266724,
0.2851542830467224
],
[
0.2310141623020172,
-0.05346527695655823,
-0.10995204746723175,
-0.037399016320705414,
-0.2227858155965805,
0.2986660301685333,
-0.07037727534770966,
0.056191567331552505,
-0.37562182545661926,
-0.07865584641695023,
0.5126975774765015,
-0.10796792805194855,
-0.4377216398715973,
-0.23079951107501984,
0.2754676640033722,
-0.38078340888023376,
0.379247784614563
],
[
-0.06051095575094223,
0.18296155333518982,
-0.07884356379508972,
-0.22479233145713806,
-0.034809600561857224,
-0.2334209382534027,
-0.26441091299057007,
-0.17463520169258118,
-0.3660305142402649,
0.006976858247071505,
-0.405804842710495,
0.15859949588775635,
-0.2280663400888443,
0.1468232423067093,
0.1943669468164444,
-0.007200213149189949,
0.2588253617286682
],
[
-0.0025153225287795067,
0.16541051864624023,
-0.22483201324939728,
0.26628783345222473,
-0.09952022135257721,
-0.3526902496814728,
0.24552574753761292,
0.24101616442203522,
0.05939788743853569,
0.24960868060588837,
0.2763345539569855,
-0.014504310674965382,
0.401587575674057,
-0.1412673145532608,
-0.20542103052139282,
-0.39227303862571716,
-0.1245405450463295
],
[
-0.01489101629704237,
-0.442043662071228,
-0.0905020609498024,
-0.10543378442525864,
-0.020450079813599586,
-0.10220003128051758,
0.17353443801403046,
-0.11333800107240677,
0.18268725275993347,
0.7068398594856262,
-0.5306816101074219,
0.09602769464254379,
0.03084583394229412,
0.29452577233314514,
0.05660795420408249,
-0.19323505461215973,
0.26553046703338623
],
[
-0.10905817896127701,
-0.0005387873388826847,
0.4620804190635681,
-0.21119607985019684,
0.09445031732320786,
-0.27449607849121094,
-0.2893728017807007,
-0.14736396074295044,
-0.12338589876890182,
0.1951233297586441,
-0.07929157465696335,
0.0520174503326416,
-0.16225674748420715,
0.11214450746774673,
-0.04820133000612259,
-0.1119072288274765,
-0.04131430387496948
],
[
-0.09867584705352783,
0.11728528887033463,
-0.06649677455425262,
-0.0717613697052002,
0.2715281546115875,
0.2710670530796051,
-0.16019245982170105,
0.18008193373680115,
-0.4145289361476898,
-0.25279751420021057,
0.20470738410949707,
0.3560670018196106,
0.272293359041214,
-0.17183461785316467,
0.0718928650021553,
-0.3431435525417328,
-0.12007743865251541
],
[
0.2646893262863159,
0.08162203431129456,
-0.062251016497612,
-0.11850492656230927,
-0.17891092598438263,
-0.44673916697502136,
-0.16576147079467773,
-0.24135617911815643,
-0.05394669249653816,
0.29974356293678284,
-0.24444086849689484,
-0.25685301423072815,
0.11704003065824509,
0.4065805971622467,
-0.22867266833782196,
-0.3498876094818115,
-0.0405145063996315
],
[
-0.1298155039548874,
-0.4707780182361603,
0.09273366630077362,
0.1623905897140503,
-0.34283575415611267,
0.16187067329883575,
0.012232727371156216,
-0.14923900365829468,
0.11674199998378754,
-0.32381942868232727,
-0.16354085505008698,
0.09329896420240402,
0.43396270275115967,
-0.13580961525440216,
0.01813303865492344,
0.07567545771598816,
0.27126699686050415
],
[
0.20082995295524597,
0.3380339741706848,
-0.40411725640296936,
0.11024849861860275,
0.17746317386627197,
0.07720617204904556,
0.4195776581764221,
0.12376824766397476,
-0.08261574804782867,
0.31845173239707947,
-0.12640881538391113,
0.04218226298689842,
0.24085019528865814,
-0.030605271458625793,
0.510686457157135,
-0.5591133236885071,
-0.3789041340351105
]
]
},
"eos_token_id": 2,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 1024,
"id2label": {
"0": "O",
"1": "B-Rating",
"2": "I-Rating",
"3": "B-Amenity",
"4": "I-Amenity",
"5": "B-Location",
"6": "I-Location",
"7": "B-Restaurant_Name",
"8": "I-Restaurant_Name",
"9": "B-Price",
"10": "B-Hours",
"11": "I-Hours",
"12": "B-Dish",
"13": "I-Dish",
"14": "B-Cuisine",
"15": "I-Price",
"16": "I-Cuisine"
},
"initializer_range": 0.02,
"intermediate_size": 4096,
"label2id": {
"B-Amenity": 3,
"B-Cuisine": 14,
"B-Dish": 12,
"B-Hours": 10,
"B-Location": 5,
"B-Price": 9,
"B-Rating": 1,
"B-Restaurant_Name": 7,
"I-Amenity": 4,
"I-Cuisine": 16,
"I-Dish": 13,
"I-Hours": 11,
"I-Location": 6,
"I-Price": 15,
"I-Rating": 2,
"I-Restaurant_Name": 8,
"O": 0
},
"layer_norm_eps": 1e-05,
"max_position_embeddings": 514,
"model_type": "roberta",
"num_attention_heads": 16,
"num_hidden_layers": 24,
"pad_token_id": 1,
"position_embedding_type": "absolute",
"torch_dtype": "float32",
"transformers_version": "4.20.1",
"type_vocab_size": 1,
"use_cache": true,
"vocab_size": 50265
}