Bert Tiny Book Text Classifier

🚀 書籍文本分類器

本項目旨在對書籍文本（主要是虛構類書籍）進行分類。

🚀 快速開始

模型詳情

模型描述

此模型基於bert-tiny進行微調，用於對書籍文本進行分類。

使用示例

基礎用法

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}]

📄 許可證

本模型使用的是MIT許可證。