CouchCat
/

ma_mlc_v7_distil

Text Classification Transformers PyTorch English distilbert multi-label Inference Endpoints

Model card Files Files and versions Community

Edit model card

Description

A Multi-label text classification model trained on a customer feedback data using DistilBert. Possible labels are:

Delivery (delivery status, time of arrival, etc.)
Return (return confirmation, return label requests, etc.)
Product (quality, complaint, etc.)
Monetary (pending transactions, refund, etc.)

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification
  
  tokenizer = AutoTokenizer.from_pretrained("CouchCat/ma_mlc_v7_distil")
  
  model = AutoModelForSequenceClassification.from_pretrained("CouchCat/ma_mlc_v7_distil")

Downloads last month: 19