Team Ai
Modelpublic

palakagl/distilbert_MultiClass_TextClassification

sourceHugging Faceupdated 5y agoView on Hugging Face
0likes10downloads
config.json162 linesDownload Raw Back to root
1{2  "_name_or_path": "AutoTrain",3  "_num_labels": 64,4  "activation": "gelu",5  "architectures": [6    "DistilBertForSequenceClassification"7  ],8  "attention_dropout": 0.1,9  "dim": 768,10  "dropout": 0.1,11  "hidden_dim": 3072,12  "id2label": {13    "0": "alarm_query",14    "1": "alarm_remove",15    "2": "alarm_set",16    "3": "audio_volume_down",17    "4": "audio_volume_mute",18    "5": "audio_volume_up",19    "6": "calendar_query",20    "7": "calendar_remove",21    "8": "calendar_set",22    "9": "cooking_recipe",23    "10": "datetime_convert",24    "11": "datetime_query",25    "12": "email_addcontact",26    "13": "email_query",27    "14": "email_querycontact",28    "15": "email_sendemail",29    "16": "general_affirm",30    "17": "general_commandstop",31    "18": "general_confirm",32    "19": "general_dontcare",33    "20": "general_explain",34    "21": "general_joke",35    "22": "general_negate",36    "23": "general_praise",37    "24": "general_quirky",38    "25": "general_repeat",39    "26": "iot_cleaning",40    "27": "iot_coffee",41    "28": "iot_hue_lightchange",42    "29": "iot_hue_lightdim",43    "30": "iot_hue_lightoff",44    "31": "iot_hue_lighton",45    "32": "iot_hue_lightup",46    "33": "iot_wemo_off",47    "34": "iot_wemo_on",48    "35": "lists_createoradd",49    "36": "lists_query",50    "37": "lists_remove",51    "38": "music_likeness",52    "39": "music_query",53    "40": "music_settings",54    "41": "news_query",55    "42": "play_audiobook",56    "43": "play_game",57    "44": "play_music",58    "45": "play_podcasts",59    "46": "play_radio",60    "47": "qa_currency",61    "48": "qa_definition",62    "49": "qa_factoid",63    "50": "qa_maths",64    "51": "qa_stock",65    "52": "recommendation_events",66    "53": "recommendation_locations",67    "54": "recommendation_movies",68    "55": "social_post",69    "56": "social_query",70    "57": "takeaway_order",71    "58": "takeaway_query",72    "59": "transport_query",73    "60": "transport_taxi",74    "61": "transport_ticket",75    "62": "transport_traffic",76    "63": "weather_query"77  },78  "initializer_range": 0.02,79  "label2id": {80    "alarm_query": 0,81    "alarm_remove": 1,82    "alarm_set": 2,83    "audio_volume_down": 3,84    "audio_volume_mute": 4,85    "audio_volume_up": 5,86    "calendar_query": 6,87    "calendar_remove": 7,88    "calendar_set": 8,89    "cooking_recipe": 9,90    "datetime_convert": 10,91    "datetime_query": 11,92    "email_addcontact": 12,93    "email_query": 13,94    "email_querycontact": 14,95    "email_sendemail": 15,96    "general_affirm": 16,97    "general_commandstop": 17,98    "general_confirm": 18,99    "general_dontcare": 19,100    "general_explain": 20,101    "general_joke": 21,102    "general_negate": 22,103    "general_praise": 23,104    "general_quirky": 24,105    "general_repeat": 25,106    "iot_cleaning": 26,107    "iot_coffee": 27,108    "iot_hue_lightchange": 28,109    "iot_hue_lightdim": 29,110    "iot_hue_lightoff": 30,111    "iot_hue_lighton": 31,112    "iot_hue_lightup": 32,113    "iot_wemo_off": 33,114    "iot_wemo_on": 34,115    "lists_createoradd": 35,116    "lists_query": 36,117    "lists_remove": 37,118    "music_likeness": 38,119    "music_query": 39,120    "music_settings": 40,121    "news_query": 41,122    "play_audiobook": 42,123    "play_game": 43,124    "play_music": 44,125    "play_podcasts": 45,126    "play_radio": 46,127    "qa_currency": 47,128    "qa_definition": 48,129    "qa_factoid": 49,130    "qa_maths": 50,131    "qa_stock": 51,132    "recommendation_events": 52,133    "recommendation_locations": 53,134    "recommendation_movies": 54,135    "social_post": 55,136    "social_query": 56,137    "takeaway_order": 57,138    "takeaway_query": 58,139    "transport_query": 59,140    "transport_taxi": 60,141    "transport_ticket": 61,142    "transport_traffic": 62,143    "weather_query": 63144  },145  "max_length": 64,146  "max_position_embeddings": 512,147  "model_type": "distilbert",148  "n_heads": 12,149  "n_layers": 6,150  "output_past": true,151  "pad_token_id": 0,152  "padding": "max_length",153  "problem_type": "single_label_classification",154  "qa_dropout": 0.1,155  "seq_classif_dropout": 0.2,156  "sinusoidal_pos_embds": false,157  "tie_weights_": true,158  "torch_dtype": "float32",159  "transformers_version": "4.15.0",160  "vocab_size": 28996161}162