Aureliano's picture
Labels update
c15a48c
{
"_name_or_path": "wn_classifier-trainer",
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0": "answer.v.01",
"1": "ask.v.01",
"2": "ask.v.02",
"3": "blow.v.01",
"4": "brandish.v.01",
"5": "break.v.05",
"6": "burn.v.01",
"7": "buy.v.01",
"8": "charge.v.17",
"9": "choose.v.01",
"10": "clean.v.01",
"11": "climb.v.01",
"12": "close.v.01",
"13": "connect.v.01",
"14": "consult.v.02",
"15": "cut.v.01",
"16": "dig.v.01",
"17": "drink.v.01",
"18": "drive.v.01",
"19": "drop.v.01",
"20": "eat.v.01",
"21": "enter.v.01",
"22": "examine.v.02",
"23": "exit.v.01",
"24": "fill.v.01",
"25": "follow.v.01",
"26": "give.v.03",
"27": "hit.v.02",
"28": "hit.v.03",
"29": "insert.v.01",
"30": "insert.v.02",
"31": "inventory.v.01",
"32": "jump.v.01",
"33": "kill.v.01",
"34": "lie_down.v.01",
"35": "light_up.v.05",
"36": "listen.v.01",
"37": "look.v.01",
"38": "lower.v.01",
"39": "memorize.v.01",
"40": "move.v.02",
"41": "note.v.04",
"42": "open.v.01",
"43": "play.v.03",
"44": "pour.v.01",
"45": "pray.v.01",
"46": "press.v.01",
"47": "pull.v.04",
"48": "push.v.01",
"49": "put.v.01",
"50": "raise.v.02",
"51": "read.v.01",
"52": "remove.v.01",
"53": "repeat.v.01",
"54": "rub.v.01",
"55": "say.v.08",
"56": "search.v.04",
"57": "sequence.n.02",
"58": "set.v.05",
"59": "shake.v.01",
"60": "shoot.v.01",
"61": "show.v.01",
"62": "sit_down.v.01",
"63": "skid.v.04",
"64": "sleep.v.01",
"65": "smash.v.02",
"66": "smell.v.01",
"67": "stand.v.03",
"68": "switch_off.v.01",
"69": "switch_on.v.01",
"70": "take.v.04",
"71": "take_off.v.06",
"72": "talk.v.02",
"73": "tell.v.03",
"74": "throw.v.01",
"75": "touch.v.01",
"76": "travel.v.01",
"77": "turn.v.09",
"78": "unknown",
"79": "unlock.v.01",
"80": "wait.v.01",
"81": "wake_up.v.02",
"82": "wear.v.02",
"83": "write.v.07"
},
"initializer_range": 0.02,
"label2id": {
"answer.v.01": 0,
"ask.v.01": 1,
"ask.v.02": 2,
"blow.v.01": 3,
"brandish.v.01": 4,
"break.v.05": 5,
"burn.v.01": 6,
"buy.v.01": 7,
"charge.v.17": 8,
"choose.v.01": 9,
"clean.v.01": 10,
"climb.v.01": 11,
"close.v.01": 12,
"connect.v.01": 13,
"consult.v.02": 14,
"cut.v.01": 15,
"dig.v.01": 16,
"drink.v.01": 17,
"drive.v.01": 18,
"drop.v.01": 19,
"eat.v.01": 20,
"enter.v.01": 21,
"examine.v.02": 22,
"exit.v.01": 23,
"fill.v.01": 24,
"follow.v.01": 25,
"give.v.03": 26,
"hit.v.02": 27,
"hit.v.03": 28,
"insert.v.01": 29,
"insert.v.02": 30,
"inventory.v.01": 31,
"jump.v.01": 32,
"kill.v.01": 33,
"lie_down.v.01": 34,
"light_up.v.05": 35,
"listen.v.01": 36,
"look.v.01": 37,
"lower.v.01": 38,
"memorize.v.01": 39,
"move.v.02": 40,
"note.v.04": 41,
"open.v.01": 42,
"play.v.03": 43,
"pour.v.01": 44,
"pray.v.01": 45,
"press.v.01": 46,
"pull.v.04": 47,
"push.v.01": 48,
"put.v.01": 49,
"raise.v.02": 50,
"read.v.01": 51,
"remove.v.01": 52,
"repeat.v.01": 53,
"rub.v.01": 54,
"say.v.08": 55,
"search.v.04": 56,
"sequence.n.02": 57,
"set.v.05": 58,
"shake.v.01": 59,
"shoot.v.01": 60,
"show.v.01": 61,
"sit_down.v.01": 62,
"skid.v.04": 63,
"sleep.v.01": 64,
"smash.v.02": 65,
"smell.v.01": 66,
"stand.v.03": 67,
"switch_off.v.01": 68,
"switch_on.v.01": 69,
"take.v.04": 70,
"take_off.v.06": 71,
"talk.v.02": 72,
"tell.v.03": 73,
"throw.v.01": 74,
"touch.v.01": 75,
"travel.v.01": 76,
"turn.v.09": 77,
"unknown": 78,
"unlock.v.01": 79,
"wait.v.01": 80,
"wake_up.v.02": 81,
"wear.v.02": 82,
"write.v.07": 83
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "single_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torch_dtype": "float32",
"transformers_version": "4.17.0",
"vocab_size": 30522
}