{ "architectures": [ "BertForSequenceClassification" ], "attention_probs_dropout_prob": 0.1, "classifier_dropout": null, "datasets": [ "imdb" ], "dtype": "float32", "gradient_checkpointing": false, "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 768, "initializer_range": 0.02, "intermediate_size": 3072, "language": "en", "layer_norm_eps": 1e-12, "license": "mit", "max_position_embeddings": 512, "metrics": [ "accuracy" ], "model-index": [ { "name": "finetuned-bert-imdb", "results": [ { "dataset": { "name": "IMDB", "type": "imdb" }, "metrics": [ { "name": "Accuracy", "type": "accuracy", "value": 0.802 } ], "task": { "name": "Text Classification", "type": "text-classification" } } ] } ], "model_type": "bert", "num_attention_heads": 12, "num_hidden_layers": 12, "pad_token_id": 0, "position_embedding_type": "absolute", "problem_type": "single_label_classification", "tags": [ "bert", "text-classification", "sentiment-analysis", "english", "imdb" ], "transformers_version": "4.57.1", "type_vocab_size": 2, "use_cache": true, "vocab_size": 30522 }