README.md

metadata

license: mit
datasets:
  - imdb
  - rotten_tomatoes
language:
  - en
metrics:
  - accuracy
pipeline_tag: text-classification

Book Test Classifier

Classify book text

Model Details

Model Description

This model is finetuned on bert-tiny for classifying book text.

• Developed by: shhossain
• Model type: [Bert]
• Language(s) (NLP): [English]
• License: [MIT]
• Finetuned from model [Bert-Tiny]: bert-tiny

Uses

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

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