CoolFace
Modelpublic

official-ak/imdb_classification_model

sourceHugging Faceapache-2.0updated 2mo agoView on Hugging Face
0likes2downloads
config.json37 linesDownload Raw Back to root
1{2  "activation": "gelu",3  "architectures": [4    "DistilBertForSequenceClassification"5  ],6  "attention_dropout": 0.1,7  "bos_token_id": null,8  "dim": 768,9  "dropout": 0.1,10  "dtype": "float32",11  "eos_token_id": null,12  "hidden_dim": 3072,13  "id2label": {14    "0": "NEGATIVE",15    "1": "POSITIVE"16  },17  "initializer_range": 0.02,18  "label2id": {19    "NEGATIVE": 0,20    "POSITIVE": 121  },22  "max_position_embeddings": 512,23  "model_type": "distilbert",24  "n_heads": 12,25  "n_layers": 6,26  "pad_token_id": 0,27  "problem_type": "single_label_classification",28  "qa_dropout": 0.1,29  "seq_classif_dropout": 0.2,30  "sinusoidal_pos_embds": false,31  "tie_weights_": true,32  "tie_word_embeddings": true,33  "transformers_version": "5.13.1",34  "use_cache": false,35  "vocab_size": 3052236}37