CoolFace
Modelpublic

EricPeter/comments-text-classification-model

sourceHugging Faceupdated 3y agoView on Hugging Face
1likes11downloads
config.json51 linesDownload Raw Back to root
1{2  "_name_or_path": "AutoTrain",3  "_num_labels": 5,4  "architectures": [5    "BertForSequenceClassification"6  ],7  "attention_probs_dropout_prob": 0.1,8  "classifier_dropout": null,9  "directionality": "bidi",10  "gradient_checkpointing": false,11  "hidden_act": "gelu",12  "hidden_dropout_prob": 0.1,13  "hidden_size": 1024,14  "id2label": {15    "0": "Excellent",16    "1": "Good",17    "2": "Neutral",18    "3": "Poor",19    "4": "Very Poor"20  },21  "initializer_range": 0.02,22  "intermediate_size": 4096,23  "label2id": {24    "Excellent": 0,25    "Good": 1,26    "Neutral": 2,27    "Poor": 3,28    "Very Poor": 429  },30  "layer_norm_eps": 1e-12,31  "max_length": 128,32  "max_position_embeddings": 512,33  "model_type": "bert",34  "num_attention_heads": 16,35  "num_hidden_layers": 24,36  "pad_token_id": 0,37  "padding": "max_length",38  "pooler_fc_size": 768,39  "pooler_num_attention_heads": 12,40  "pooler_num_fc_layers": 3,41  "pooler_size_per_head": 128,42  "pooler_type": "first_token_transform",43  "position_embedding_type": "absolute",44  "problem_type": "single_label_classification",45  "torch_dtype": "float32",46  "transformers_version": "4.20.0",47  "type_vocab_size": 2,48  "use_cache": true,49  "vocab_size": 2899650}51