sentiment-model / config.json
dancingninjas's picture
Update config.json
9459299
raw
history blame contribute delete
802 Bytes
{
"_name_or_path": "dancingninjas/sentiment-model",
"activation": "gelu",
"architectures": [
"TFAutoModelForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"initializer_range": 0.02,
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "single_label_classification",
"seq_classif_dropout": 0.2,
"transformers_version": "4.26.1",
"vocab_size": 30522,
"description": "A fine-tuned DistilBERT model for sentiment analysis.",
"tags": ["text-classification", "sentiment-analysis"],
"pipeline_tag": "text-classification",
"framework": "transformers",
"pytorch_version": "1.8.1",
"tensorflow_version": "2.7.0",
"license": "MIT"
}