resume_sorter / config.json
Kanakmi's picture
Upload TFDistilBertForSequenceClassification
d91c764
raw
history blame
1.84 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": "Data Science",
"1": "HR",
"2": "Advocate",
"3": "Arts",
"4": "Web Designing",
"5": "Mechanical Engineer",
"6": "Sales",
"7": "Health and fitness",
"8": "Civil Engineer",
"9": "Java Developer",
"10": "Business Analyst",
"11": "SAP Developer",
"12": "Automation Testing",
"13": "Electrical Engineering",
"14": "Operations Manager",
"15": "Python Developer",
"16": "DevOps Engineer",
"17": "Network Security Engineer",
"18": "PMO",
"19": "Database",
"20": "Hadoop",
"21": "ETL Developer",
"22": "DotNet Developer",
"23": "Blockchain",
"24": "Testing"
},
"initializer_range": 0.02,
"label2id": {
"Advocate": 2,
"Arts": 3,
"Automation Testing": 12,
"Blockchain": 23,
"Business Analyst": 10,
"Civil Engineer": 8,
"Data Science": 0,
"Database": 19,
"DevOps Engineer": 16,
"DotNet Developer": 22,
"ETL Developer": 21,
"Electrical Engineering": 13,
"HR": 1,
"Hadoop": 20,
"Health and fitness": 7,
"Java Developer": 9,
"Mechanical Engineer": 5,
"Network Security Engineer": 17,
"Operations Manager": 14,
"PMO": 18,
"Python Developer": 15,
"SAP Developer": 11,
"Sales": 6,
"Testing": 24,
"Web Designing": 4
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"transformers_version": "4.25.1",
"vocab_size": 30522
}