Instructions to use dima806/toxic-comments-classifier-distilbert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dima806/toxic-comments-classifier-distilbert with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="dima806/toxic-comments-classifier-distilbert")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("dima806/toxic-comments-classifier-distilbert") model = AutoModelForSequenceClassification.from_pretrained("dima806/toxic-comments-classifier-distilbert") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -3,5 +3,7 @@ license: apache-2.0
|
|
| 3 |
metrics:
|
| 4 |
- accuracy
|
| 5 |
- f1
|
|
|
|
|
|
|
| 6 |
---
|
| 7 |
See https://www.kaggle.com/code/dima806/toxic-comments-weighted-classify-distilbert for more details.
|
|
|
|
| 3 |
metrics:
|
| 4 |
- accuracy
|
| 5 |
- f1
|
| 6 |
+
base_model:
|
| 7 |
+
- google-bert/bert-base-uncased
|
| 8 |
---
|
| 9 |
See https://www.kaggle.com/code/dima806/toxic-comments-weighted-classify-distilbert for more details.
|