Peterard's picture
Create README.md
3890f42
|
raw
history blame
650 Bytes
metadata
language:
  - en
tags:
  - text-classification
widget:
  - text: Please add a like button!
    example_title: Likely feature request
  - text: This thing is always crashing
    example_title: Unlikely feature request

How to use this classifier:

from transformers import pipeline

pipe = pipeline("text-classification", model="Peterard/distilbert_feature_classifier")

pipe("Please add a like button!")
# [{'label': 'feature_request', 'score': 0.8930749893188477}]

pipe("This thing is always crashing")
#[{'label': 'no_feature_request', 'score': 0.9975591897964478}]

N.B. The label will change depending on which is the likelier class