hfunakura/bert-feedback-classifier

What

製品に対するユーザーからのフィードバックを改善要望と感想の二値に振り分ける分類モデルです。 bert-base-japaneseをファインチューニングして作成しています。

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification
from transformers import pipeline

tokenizer = AutoTokenizer.from_pretrained("hfunakura/bert-feedback-classifier")
model = AutoModelForSequenceClassification.from_pretrained("hfunakura/bert-feedback-classifier")

classifier = pipeline(
    "text-classification",
    model=model,
    tokenizer=tokenizer
)

classifier("アプリが頻繁にクラッシュするので使いづらいです。")