Instructions to use themaverick1/goodreads-genre-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use themaverick1/goodreads-genre-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="themaverick1/goodreads-genre-classifier")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("themaverick1/goodreads-genre-classifier") model = AutoModelForSequenceClassification.from_pretrained("themaverick1/goodreads-genre-classifier") - Notebooks
- Google Colab
- Kaggle
| { | |
| "num_train_epochs": 2, | |
| "per_device_train_batch_size": 4, | |
| "learning_rate": 5e-05, | |
| "eval_steps": 200 | |
| } |