belief_detection_model / config.json
junaid1993's picture
Create config.json
f909b21 verified
raw
history blame contribute delete
752 Bytes
{
"model_type": "distilbert",
"model_name": "distilbert-base-custom",
"description": "A custom fine-tuned DistilBERT model for belief detection task.",
"task": "Belief Detection",
"tags": ["distilbert", "belief-detection", "NLP"],
"language": "english",
"license": "Apache-2.0",
"framework": "PyTorch",
"transformers_version": "4.11.3",
"inputs": {
"text": "A piece of text to classify."
},
"outputs": {
"class": "Predicted class label (0: Strong, 1: Moderate, 2: Weak, 3: No Beliefs Expressed)"
},
"label_to_id": {
"Strong": 0,
"Moderate": 1,
"Weak": 2,
"No Beliefs Expressed": 3
},
"id_to_label": {
"0": "Strong",
"1": "Moderate",
"2": "Weak",
"3": "No Beliefs Expressed"
}
}