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许可证。