File size: 752 Bytes
f909b21 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
{
"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"
}
}
|