hfunakura's picture
Update README.md
7cbe68d verified
|
raw
history blame
907 Bytes
metadata
library_name: transformers
tags:
  - business
license: apache-2.0
language:
  - ja
metrics:
  - accuracy
base_model:
  - tohoku-nlp/bert-base-japanese
pipeline_tag: text-classification

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("アプリが頻繁にクラッシュするので使いづらいです。")