Hugging Face's logo

CouchCat
/
ma_mlc_v7_distil

Text Classification
Transformers
PyTorch
English
distilbert
multi-label
Inference Endpoints
Model card Files Community
1

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
10
Inference Examples
Text Classification
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.