bert_model / config.json
DRAGOO's picture
Update config.json
b7ca4f1 verified
raw
history blame
2.93 kB
{
"_name_or_path": "distilbert-base-uncased",
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {'0': 'AIR COMPRESSOR',
'1': 'ALLEN KEY',
'2': 'ANGLE GAUGE',
'3': 'AXE',
'4': 'BACK SAW',
'5': 'BAND SAW',
'6': 'BONE SAW',
'7': 'CALIPER',
'8': 'CHISEL',
'9': 'CIRCULAR SAW',
'10': 'CLAMP',
'11': 'COPING SAW',
'12': 'CORDLESS DRILL',
'13': 'CROSSCUT SAW',
'14': 'DRILLS',
'15': 'FILE',
'16': 'FRET SAW',
'17': 'GARAGE CABINET',
'18': 'GAUGE',
'19': 'GRINDER',
'20': 'HACK SAW',
'21': 'HAMMER',
'22': 'HAND SAW',
'23': 'HEAT GUN',
'24': 'IMPACT DRIVER',
'25': 'IMPACT WRENCH',
'26': 'INCLINOMETER',
'27': 'KEYHOLE SAW',
'28': 'LASER LEVEL',
'29': 'LATHE',
'30': 'LEVEL',
'31': 'MALLET',
'32': 'MICROMETER',
'33': 'MITER SAW',
'34': 'NAIL GUN',
'35': 'NUT DRIVER',
'36': 'PLANE',
'37': 'PLIERS',
'38': 'PROTRACTOR',
'39': 'PRUNING SAW',
'40': 'RASP',
'41': 'RECIPROCATING SAW',
'42': 'ROLLING TOOL STORAGE',
'43': 'SANDER',
'44': 'SCREW DRIVERS',
'45': 'STAPLER',
'46': 'TABLE SAW',
'47': 'TAPE MEASURE',
'48': 'TILE SAW',
'49': 'TOOL BAG',
'50': 'TOOL BELT',
'51': 'TOOL BOX',
'52': 'TOOL CABINET',
'53': 'TOOL CHEST',
'54': 'VENEER SAW',
'55': 'WRENCH'},
"initializer_range": 0.02,
"label2id": {
"AIR COMPRESSOR" : "0" ,
"ALLEN KEY" : "1" ,
"ANGLE GAUGE" : "2" ,
"AXE" : "3" ,
"BACK SAW" : "4" ,
"BAND SAW" : "5" ,
"BONE SAW" : "6" ,
"CALIPER" : "7" ,
"CHISEL" : "8" ,
"CIRCULAR SAW" : "9" ,
"CLAMP" : "10" ,
"COPING SAW" : "11" ,
"CORDLESS DRILL" : "12" ,
"CROSSCUT SAW" : "13" ,
"DRILLS" : "14" ,
"FILE" : "15" ,
"FRET SAW" : "16" ,
"GARAGE CABINET" : "17" ,
"GAUGE" : "18" ,
"GRINDER" : "19" ,
"HACK SAW" : "20" ,
"HAMMER" : "21" ,
"HAND SAW" : "22" ,
"HEAT GUN" : "23" ,
"IMPACT DRIVER" : "24" ,
"IMPACT WRENCH" : "25" ,
"INCLINOMETER" : "26" ,
"KEYHOLE SAW" : "27" ,
"LASER LEVEL" : "28" ,
"LATHE" : "29" ,
"LEVEL" : "30" ,
"MALLET" : "31" ,
"MICROMETER" : "32" ,
"MITER SAW" : "33" ,
"NAIL GUN" : "34" ,
"NUT DRIVER" : "35" ,
"PLANE" : "36" ,
"PLIERS" : "37" ,
"PROTRACTOR" : "38" ,
"PRUNING SAW" : "39" ,
"RASP" : "40" ,
"RECIPROCATING SAW" : "41" ,
"ROLLING TOOL STORAGE" : "42" ,
"SANDER" : "43" ,
"SCREW DRIVERS" : "44" ,
"STAPLER" : "45" ,
"TABLE SAW" : "46" ,
"TAPE MEASURE" : "47" ,
"TILE SAW" : "48" ,
"TOOL BAG" : "49" ,
"TOOL BELT" : "50" ,
"TOOL BOX" : "51" ,
"TOOL CABINET" : "52" ,
"TOOL CHEST" : "53" ,
"VENEER SAW" : "54" ,
"WRENCH" : "55"
},
"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.40.0",
"vocab_size": 30522
}