Text Classification
Transformers
English
bert
minbert
transformer
sentiment
tokenizer
classification
Instructions to use GlowCheese/minBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use GlowCheese/minBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="GlowCheese/minBERT")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("GlowCheese/minBERT", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| from constants import * | |
| from types import SimpleNamespace | |
| from classifier import classifier_run | |
| ARGUMENTS = SimpleNamespace( | |
| dataset='sst', | |
| batch_size=BATCH_SIZE_SST, | |
| train=IDS_SST_TRAIN, | |
| dev=IDS_SST_DEV, | |
| test=IDS_SST_TEST, | |
| lr=1e-3, | |
| fine_tune_mode='last-linear-layer' | |
| ) | |
| classifier_run(ARGUMENTS) |