Team Ai
Modelpublic

palakagl/Roberta_Multiclass_TextClassification

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