attention-sentiment-classifier / model-index.json
ericwei's picture
Initial model upload
f3ff1d7 verified
{
"name": "attention-sentiment-classifier",
"description": "Attention-based sentiment classification model that visualizes which parts of text influence predictions",
"tags": [
"pytorch",
"text-classification",
"sentiment-analysis",
"attention-mechanism",
"english",
"sst2"
],
"license": "gpl-3.0",
"library_name": "transformers",
"pipeline_tag": "text-classification"
}