CoolFace
Modelpublic

PavanDeepak/Topic_Classification

sourceHugging Facemitupdated 3y agoView on Hugging Face
0likes14downloads
config.json56 linesDownload Raw Back to root
1{2  "_name_or_path": "text-classification-model",3  "architectures": [4    "BertForSequenceClassification"5  ],6  "attention_probs_dropout_prob": 0.1,7  "classifier_dropout": null,8  "gradient_checkpointing": false,9  "hidden_act": "gelu",10  "hidden_dropout_prob": 0.1,11  "hidden_size": 768,12  "id2label": {13    "0": "Food",14    "1": "Videogames & Shows",15    "2": "Kids and fun",16    "3": "Homestyle",17    "4": "Travel",18    "5": "Health",19    "6": "Charity",20    "7": "Electronics & Technology",21    "8": "Sports",22    "9": "Cultural & Music",23    "10": "Education",24    "11": "Convenience"25  },26  "initializer_range": 0.02,27  "intermediate_size": 3072,28  "label2id": {29    "Charity": 6,30    "Convenience": 11,31    "Cultural & Music": 9,32    "Education": 10,33    "Electronics & Technology": 7,34    "Food": 0,35    "Health": 5,36    "Homestyle": 3,37    "Kids and fun": 2,38    "Sports": 8,39    "Travel": 4,40    "Videogames & Shows": 141  },42  "layer_norm_eps": 1e-12,43  "max_position_embeddings": 512,44  "model_type": "bert",45  "num_attention_heads": 12,46  "num_hidden_layers": 12,47  "pad_token_id": 0,48  "position_embedding_type": "absolute",49  "problem_type": "single_label_classification",50  "torch_dtype": "float32",51  "transformers_version": "4.37.0",52  "type_vocab_size": 2,53  "use_cache": true,54  "vocab_size": 3052255}56