Instructions to use unitary/toxic-bert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use unitary/toxic-bert with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="unitary/toxic-bert")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("unitary/toxic-bert") model = AutoModelForSequenceClassification.from_pretrained("unitary/toxic-bert") - Inference
- Notebooks
- Google Colab
- Kaggle
Training of toxic-bert
#3
by JOhan1906 - opened
Hi,
How was toxic bert trained? Were all the layers in the pre-trained Bert model frozen except for the last two?
i wan to finetune toxic-bert
using autotrain
can someone tell me what should be there in Column mapping