Bert Tiny Book Text Classifier

🚀 書籍テキスト分類器

このモデルは、主に架空の書籍を含む書籍テキストを分類することができます。

✨ 主な機能

• 書籍テキスト（主に架空の書籍）を分類することができます。

📦 インストール

このREADMEには具体的なインストール手順が記載されていないため、このセクションは省略されます。

💻 使用例

基本的な使用法

from transformers import pipeline, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('prajjwal1/bert-tiny')

pipe = pipeline('text-classification', model='shhossain/bert-tiny-book-text-classifier')

book_text = """Shen Yuanye didn’t expect to go back to Huang Ni’s matter again, this matter of being killed without finding the murderer, who else could be beside her."""

pipe(book_text) # LABEL_1
>> [{'label': 'LABEL_1', 'score': 0.9998537302017212}]

normal_text = """I am so sorry this is a day late, guys. Unfortunately, my internet was down so it was out of my control. Its still intermittent but hopefully it will be fine by next week. Hopefully the fact that Skye and Pietro are back in form will help make up for it."""

pipe(normal_text) # LABEL_0
>> [{'label': 'LABEL_0', 'score': 0.9984021782875061}]

📚 ドキュメント

モデルの詳細

モデルの説明

このモデルは、書籍テキストを分類するためにbert - tinyをファインチューニングしたものです。

📄 ライセンス

このモデルはMITライセンスの下で提供されています。