Views
No views yet
改善要望と感想の二値に振り分ける分類モデルです。
bert-base-japaneseをファインチューニングして作成しています。1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2from transformers import pipeline
3
4tokenizer = AutoTokenizer.from_pretrained("hfunakura/bert-feedback-classifier")
5model = AutoModelForSequenceClassification.from_pretrained("hfunakura/bert-feedback-classifier")
6
7classifier = pipeline(
8 "text-classification",
9 model=model,
10 tokenizer=tokenizer
11)
12
13classifier("アプリが頻繁にクラッシュするので使いづらいです。")